Estrazione del genere principale

Ti sei unito al team dati di una startup di streaming di film. Memorizzano i metadati dei film in un file Parquet, dove per ogni film genres è una lista di stringhe, perché la maggior parte dei titoli copre più generi. Estrai il primo genere elencato da ogni riga in una colonna primary_genre pulita a livello superiore.

polars è caricato come pl e il DataFrame movies è stato precaricato dal file Parquet.

Questo esercizio fa parte del corso

Scalare e ottimizzare le pipeline di dati con Polars

Visualizza corso

Istruzioni dell'esercizio

Estrai il primo elemento dalla lista genres di ciascuna riga usando un'espressione per liste.
Assegna alla nuova colonna l'alias primary_genre.

esercizio interattivo pratico

Prova questo esercizio completando questo codice di esempio.

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

Modifica ed esegui il codice