Как преобразовать столбец в кадре данных Polars из строки в Enum?Python

Программы на Python
Anonymous
Как преобразовать столбец в кадре данных Polars из строки в Enum?

Сообщение Anonymous »

Я читаю простой файл данных, состоящий из двух столбцов, кода и метки, как показано ниже:

1=Розничная торговля2=Приготовление
3=Домашняя инфузия
4=Институциональный
5 =LTC Pharm

используя этот код:
df = pl.read_csv(refdir + '/RawData/PharmacyType.csv', columns=None,
new_columns=['PharmacyType', 'PharmacyType_lbl'], separator='=')

Я хотел бы преобразовать текст в тип Enum. Мой код кажется логичным
listLbl = df.select('PharmacyType_lbl').to_series().to_list()

enumPharmType = pl.Enum(listLbl)

Я хотел бы добавить третий столбец в фрейм данных, который преобразует строку в тип Enum.
df = df.with_columns(pl.col('PharmacyType_lbl').cast(pl.enumPharmType).alias('EnumLabel'))

Однако я получаю сообщение об ошибке
module 'polars' has no attribute 'enumPharmType'


Подробнее здесь: https://stackoverflow.com/questions/786 ... ng-to-enum

Вернуться в «Python»