Получить ячейку столбца DataFrame Enum как EnumPython

Программы на Python
Ответить
Anonymous
 Получить ячейку столбца DataFrame Enum как Enum

Сообщение Anonymous »

У меня есть DataFrame, в котором несколько столбцов определены как Enums. Когда я получаю ячейку в столбце Enum, значение возвращается в виде строки. Как получить значение в виде Enum? Ниже показан небольшой репродукция:
import polars as pl

flags: pl.Enum = pl.Enum(["foo", "bar", "baz"])
df: pl.DataFrame = pl.DataFrame(data={"x": ["foo", "bar", "baz"]}, schema_overrides={"x": flags})
df.dtypes
# Returns [Enum(categories=['foo', 'bar', 'baz'])]
type(df[0, "x"])
# Returns str instead of Enum
# Series.to_list() returns list of str instead of list of Enum
[type(t) for t in df[:, "x"].to_list()]
# Returns [str, str, str]


Подробнее здесь: https://stackoverflow.com/questions/791 ... as-an-enum
Ответить

Быстрый ответ

Изменение регистра текста: 
Смайлики
:) :( :oops: :roll: :wink: :muza: :clever: :sorry: :angel: :read: *x)
Ещё смайлики…
   
К этому ответу прикреплено по крайней мере одно вложение.

Если вы не хотите добавлять вложения, оставьте поля пустыми.

Максимально разрешённый размер вложения: 15 МБ.

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