Polars中实现zfill字符串补零的正确方法是什么

现有map_elements写法的适用性 #

如果你对性能没有高要求，数据规模也不大，这个写法完全能用，没有什么不合适的。
唯一要留意的是空值处理：如果列里存在None值，当前lambda会把空值转成字符串"None"再执行zfill，和预期不符的话记得提前做空值处理，或者在lambda里加个判空分支。

更优的原生实现方案 #

Polars目前确实没有直接内置同名的zfill方法，但用内置的原生字符串表达式完全可以拼出等价逻辑，计算全程跑在Rust内核里，比Python层的map_elements性能高几个量级，也不会有跨语言调用、类型推断出错的问题：

• 如果你处理的内容都是非负数值/字符串，没有负号开头的条目，直接用左填充接口就能实现完全一致的zfill效果：

  # 等价于非负场景下 str(x).zfill(5)
  pl.col("column").cast(pl.String).str.pad_start(5, "0")
  

• 如果列里存在带负号的数字，需要和Python原生zfill的行为完全对齐（即负号保留在最左侧，补0放在负号之后，比如"-12".zfill(5)返回"-0012"），可以用分支表达式做兼容：

  str_col = pl.col("column").cast(pl.String)
  pl.when(str_col.str.starts_with("-"))
    .then("-" + str_col.str.slice(1).str.pad_start(4, "0"))
    .otherwise(str_col.str.pad_start(5, "0"))
  

如果要频繁用到zfill逻辑，完全可以把上面的逻辑封装成可复用的自定义表达式，传入目标填充长度作为参数即可，不管是复用性还是稳定性都比调Python lambda好。

内容的提问来源于stack exchange，提问作者cnpryer