Extragerea genului principal

Tocmai ai intrat în echipa de date a unui startup de streaming de filme. Metadatele filmelor sunt stocate într-un fișier Parquet, unde câmpul genres al fiecărui film este o listă de șiruri de caractere – deoarece majoritatea filmelor aparțin mai multor genuri. Extrage primul gen din fiecare rând într-o coloană separată, primary_genre, la nivelul principal al DataFrame-ului.

polars este importat ca pl, iar DataFrame-ul movies este deja încărcat din fișierul Parquet.

Acest exercițiu face parte din cursul

Scalarea și optimizarea pipeline-urilor de date cu Polars

Vezi cursul

Instrucțiuni pentru exercițiu

Extrage primul element din lista genres a fiecărui rând folosind o expresie de tip list.
Denumește noua coloană primary_genre cu ajutorul unui alias.

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

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

Editează și rulează codul