Создание столбца типа Enum

У фильмов есть фиксированный набор статусов: Released, Rumored, Post Production и другие. Поскольку допустимые значения известны заранее, тип Enum подходит здесь лучше, чем Categorical: он добавляет проверку, которая обнаруживает любые неизвестные статусы до того, как они нарушат работу пайплайна.

Набор данных movies по-прежнему доступен, а также заранее определённый объект status_enum, содержащий все допустимые значения.

Это упражнение является частью курса

Масштабирование и оптимизация конвейеров данных с Polars

Посмотреть курс

Инструкции к упражнению

Приведите столбец status к типу, используя заранее определённый объект status_enum.

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

result = movies.with_columns(
    # Cast to the enum
    pl.col("____").____(____)
).select("movie_title", "status").head(8)
print(result)

Редактировать и запускать код