覆寫推斷的綱要(schema)
當 Polars 掃描供應商匯出的檔案時,會用前幾列來推斷欄位型別。branch_code 欄位使用包含前置零的 3 位數代碼("001"、"002"、…),但 Polars 將其視為整數並解析為 Int64,前置零就被悄悄去掉了。請覆寫綱要,讓 branch_code 保持為字串。
已經為你印出了推斷的綱要,你可以在未覆寫前先看看 Polars 的推測。
本練習屬於課程
使用 Polars 擴充與最佳化資料管線
練習說明
- 覆寫
branch_code,讓它以pl.String讀取。
動手互動練習
試著完成這個範例程式碼,體驗一下這個練習。
schema = pl.scan_csv(
MESSY_CSV_PATH,
separator=";",
skip_rows=2,
infer_schema_length=5,
# Force branch_code to String
schema_overrides={"____": pl.____},
).collect_schema()
print("\nOverridden schema:")
print(schema)