Извлечение основного жанра

Вы присоединились к команде данных стартапа по стриминговым фильмам. Метаданные фильмов хранятся в файле Parquet, где поле genres каждого фильма представляет собой список строк — ведь большинство картин относится сразу к нескольким жанрам. Извлеките первый жанр из каждой строки в отдельный столбец верхнего уровня primary_genre.

polars загружен как pl, а DataFrame movies предварительно загружен из файла Parquet.

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

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

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

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

Извлеките первый элемент из списка genres в каждой строке с помощью list-выражения.
Задайте новому столбцу псевдоним primary_genre.

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

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

result = movies.select(
    "movie_title",
    "genres",
    # Get the first genre
    pl.col("genres").list.____(0).alias("____"),
).head(8)
print(result)

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