Полярные: удалить подстроку из столбца на основе другого столбца. ⇐ Python

[Anonymous]

Есть ли какая-либо оптимизация на основе Polars, которую можно применить к методологии apply-lambda в этом посте. Удалить подстроку из столбца на основе другого столбца?
В кадре данных Polars ниже, как мы можем удалить "_sub" подстроки на основе значения sub?
import polars as pl

pl.DataFrame(
{"origin": ["id1_COUNTRY", "id2_NAME"],
"sub": ["COUNTRY", "NAME"]}
)

shape: (2, 2)
┌─────────────┬─────────┐
│ origin ┆ sub │
│ --- ┆ --- │
│ str ┆ str │
╞═════════════╪═════════╡
│ id1_COUNTRY ┆ COUNTRY │
│ id2_NAME ┆ NAME │
└─────────────┴─────────┘

Ожидаемый результат должен выглядеть так:
shape: (2, 3)
┌─────────────┬─────────┬─────┐
│ origin ┆ sub ┆ out │
│ --- ┆ --- ┆ --- │
│ str ┆ str ┆ str │
╞═════════════╪═════════╪═════╡
│ id1_COUNTRY ┆ COUNTRY ┆ id1 │
│ id2_NAME ┆ NAME ┆ id2 │
└─────────────┴─────────┴─────┘


Подробнее здесь: https://stackoverflow.com/questions/783 ... her-column