Витягнення основного жанру

Ви приєдналися до команди даних у стартапі стримінґових фільмів. Вони зберігають метадані фільмів у файлі Parquet, де genres для кожного фільму — це список рядків, адже більшість фільмів належать до кількох жанрів. Витягніть перший зазначений жанр з кожного рядка в окремий «плоский» стовпець верхнього рівня primary_genre.

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

Ця вправа є частиною курсу

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

Переглянути курс

Інструкції до вправи

Витягніть перший елемент зі списку genres у кожному рядку, використавши вираз для списків.
Додайте псевдонім для нового стовпця як primary_genre.

Інтерактивна практична вправа

Спробуйте виконати цю вправу, доповнивши цей зразок коду.

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

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