Полярные значения: используйте значение из столбца в качестве имени столбца в выражении When then ⇐ Python

[Anonymous]

В кадре данных Polars у меня есть столбец, содержащий имена других столбцов (столбец «id_column_name»). Я хочу использовать эти имена в выражении «когда-то» с помощью pl.col(), чтобы создать новый столбец («id»), который получает свои значения из этих других столбцов («id_column1», «id_column2»). Каждая строка может получить значение из другого столбца в df.

Код: Выделить всё

# initial df
df = pl.DataFrame({
'id_column1': [123, 456],
'id_column2': ['abc', 'def'],
'id_column_name': ['id_column1', 'id_column2']
})

# required output df
df_out = pl.DataFrame({
'id_column1': [123, 456],
'id_column2': ['abc', 'def'],
'id_column_name': ['id_column1', 'id_column2'],
'id': ['123', 'def']
})

# one of the trings I tried
df = df.with_columns(
pl.when(pl.col('id_column_name').is_not_null())
.then(pl.col(pl.col('id_column_name')))
.otherwise(None)
.cast(pl.String)
.alias('id')
)

Это приводит к ошибке: Ожидаемая строка или DataType получила значение «Expr».

Использование str() или .str для преобразования выражения в обычную строку приводит к другим ошибкам:

Ожидается str или DataType, получило «ExprStringNameSpace».

Это не может быть так сложно, не так ли?

Подробнее здесь: https://stackoverflow.com/questions/791 ... expression