開始使用免費開始

覆寫推斷的綱要(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)
編輯並執行程式碼