Wyodrębnianie głównego gatunku

Dołączasz do zespołu analitycznego w startupie streamingowym. Metadane filmów są przechowywane w pliku Parquet, gdzie kolumna genres każdego filmu zawiera listę łańcuchów znaków — większość filmów należy bowiem do kilku gatunków. Wyciągnij pierwszy gatunek z każdego wiersza i umieść go w nowej kolumnie najwyższego poziomu o nazwie primary_genre.

polars jest załadowany jako pl, a DataFrame movies jest wczytany z pliku Parquet.

To ćwiczenie jest częścią kursu

Skalowanie i optymalizacja potoków danych w Polars

Zobacz kurs

Instrukcje do ćwiczenia

Wyodrębnij pierwszy element z listy genres w każdym wierszu, korzystając z wyrażenia listowego.
Nadaj nowej kolumnie alias primary_genre.

Interaktywne ćwiczenie praktyczne

Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.

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

Edytuj i uruchom kod