Extraire le genre principal

Vous venez de vous joindre à l'équipe de données d'une jeune entreprise de diffusion de films en continu. 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, puisque la plupart des films couvrent plusieurs genres. Extrayez le premier genre indiqué de chaque ligne dans une colonne de niveau supérieur propre nommée primary_genre.

polars est importé sous pl, et le DataFrame movies est déjà chargé à partir du fichier Parquet.

Cette activité fait partie du cours

Mise à l'échelle et optimisation des pipelines de données avec Polars

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