Créer une colonne Categorical

Revenons au jeu de données de films de la startup de streaming. La plupart des lignes de original_language utilisent les mêmes quelques codes (en, fr, es, …). Le transtypage en Categorical encode chaque étiquette unique en un entier en arrière-plan, ce qui économise de la mémoire pour les chaînes répétées.

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

Convertissez la colonne original_language en pl.Categorical.

Exercice interactif pratique

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

movies_cat = movies.with_columns(
    # Cast to Categorical to save memory
    pl.col("original_language").____(pl.____)
)

result = movies_cat.select("movie_title", "original_language").head(8)
print(result)

Modifier et exécuter le code