Enum 列の作成
映画のステータスは Released、Rumored、Post Production など、あらかじめ決まった値のセットを持っています。使用できる値が事前にわかっている場合、Categorical よりも Enum の方が適しています。Enum を使うと、未知のステータスがパイプラインに混入する前に検出できるバリデーション機能が追加されます。
movies は引き続き利用できます。また、すべての許容値を列挙した status_enum もあらかじめ定義されています。
この演習はコースの一部です
Polars によるデータパイプラインのスケーリングと最適化
演習の手順
- あらかじめ定義された
status_enumを使って、status列をキャストしましょう。
実践的なインタラクティブ演習
このサンプルコードを完成させて、この演習に挑戦してみましょう。
result = movies.with_columns(
# Cast to the enum
pl.col("____").____(____)
).select("movie_title", "status").head(8)
print(result)