Заявление о деле Polars ⇐ Python

[Anonymous]

Я пытаюсь перенять полярные свойства пакета из Python.
Я имею опыт работы с R, поэтому понимаю, что это может быть невероятно простой вопрос.
Я хочу реализовать оператор случая, где, если какое-либо из приведенных ниже условий истинно, оно помечается как 1, в противном случае оно будет равно 0. Мой новый столбец будет называться «my_new_column_flag».
Однако я получаю сообщение об ошибке

Traceback (последний вызов — последний):
Файл "", строка 2, в
Файл "C:\Users" \foo\Miniconda3\envs\env\lib\site-packages\polars\internals\lazy_functions.py", строка 204, в столбце
return pli.wrap_expr(pycol(name))
TypeError: аргумент 'name': объект 'int' невозможно преобразовать в 'PyString'

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

import polars as pl
import numpy as np

np.random.seed(12)

df = pl.DataFrame(
{
"nrs": [1, 2, 3, None, 5],
"names": ["foo", "ham", "spam", "egg", None],
"random": np.random.rand(5),
"groups": ["A", "A", "B", "C", "B"],
}
)
print(df)

df.with_columns(
pl.when(pl.col('nrs') == 1).then(pl.col(1))
.when(pl.col('names') == 'ham').then(pl.col(1))
.when(pl.col('random') == 0.014575).then(pl.col(1))
.otherwise(pl.col(0))
.alias('my_new_column_flag')
)

Кто-нибудь может помочь?


Подробнее здесь: https://stackoverflow.com/questions/753 ... -statement