Extrahera ett primärt genre
Du har börjat på datateamet på en streamingtjänst för film. Filmmetadata lagras i en Parquet-fil där varje films genres är en lista med strängar, eftersom de flesta filmer tillhör flera genrer. Din uppgift är att plocka ut det första genret från varje rad och lägga det i en ny toppnivåkolumn kallad primary_genre.
polars är inläst som pl, och DataFramen movies är förinstansierad från Parquet-filen.
Den här övningen är en del av kursen
Skalning och optimering av datapipelines med Polars
Övningsinstruktioner
- Extrahera det första elementet från
genres-listan i varje rad med ett listuttryck. - Ge den nya kolumnen aliaset
primary_genre.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
result = movies.select(
"movie_title",
"genres",
# Get the first genre
pl.col("genres").list.____(0).alias("____"),
).head(8)
print(result)