Extraire un genre principal

Vous rejoignez l'équipe data d'une startup de streaming de films. Les métadonnées des films sont stockées dans un fichier Parquet, où le champ genres de chaque film est une liste de chaînes de caractères, car la plupart des films couvrent plusieurs genres. Extrayez le premier genre listé de chaque ligne dans une colonne de premier niveau propre nommée primary_genre.

polars est importé sous le nom pl, et le DataFrame movies est préchargé depuis le fichier Parquet.

Cet exercice fait partie du cours

<cours>Mise à l'échelle et optimisation des pipelines de données avec Polars</cours>

Voir le cours

Instructions de l’exercice

Extrayez le premier élément de la liste genres de chaque ligne à l'aide d'une expression de liste.
Donnez à la nouvelle colonne l'alias primary_genre.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

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

Modifier et exécuter le code