覆盖推断的模式
当 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)