本文深入解析如何在 Polars 数据处理框架中，不依赖低效的 map_rows 或 Python 循环，而是利用其强大的原生表达式 API，实现根据另一列动态指定的精度，对数值列进行高性能、逐行的四舍五入操作。
在数据分析与清洗的实际工作中，你是否面临过这样的挑战：需要依据数据表中另一列（例如“有效数字位数”或“精度要求”）动态变化的值，来对目标数值列进行逐行、不同精度的四舍五入？如果直接采用 `map_rows` 或 `apply` 方法，会立即将计算拖入 Python 解释器的单线程循环，使得 Polars 引以为傲的向量化计算与并行处理优势完全丧失。在处理海量数据时，这种性能损耗是不可接受的。
幸运的是，Polars 强大的原生表达式 API 为我们提供了两种纯向量化、零 Python 开销的优雅解决方案：数学缩放法与条件分支聚合法。下面我们将对这两种方法进行详细剖析与对比。
✅ 方案一：数学缩放法（推荐｜通用性最佳）
此方案的核心逻辑极为巧妙，可概括为“按精度放大、取整、再按精度缩小”。具体而言，先将原始浮点数乘以 \(10^{\text{精度位数}}\)，将需要保留的小数位移至整数部分；接着调用 Polars 内置的 `.round()` 方法（默认执行四舍五入至整数）；最后再除以 \(10^{\text{精度位数}}\)，将数值还原至正确的量级。整个过程完全基于 Polars 的高效算术表达式，无任何条件分支，性能表现最优。
import polars as pl
df = df.with_columns(
(
pl.col("reverse_rate_from_euro")
* pl.lit(10).pow(pl.col("sig_figs_len"))
).round()
* pl.lit(0.1).pow(pl.col("sig_figs_len"))
.alias("reverse_rate_to_euro_rounded_sig_figs")
)
这里有一个性能细节：使用 `pl.lit(0.1)` 作为 \(10^{-1}\) 的等价写法，比 `1 / 10.pow(...)` 更为高效。此方案支持 `u32` 或 `i64` 类型的精度位列，并且对于 NaN、Inf 等特殊值的处理行为，与 Polars 标准 `round()` 函数完全一致，确保了结果的可靠性。
✅ 方案二：条件分支 + coalesce()（适用于精度值离散且种类少）
如果你的 `sig_figs_len`（精度位数）列取值是离散的，且唯一值数量非常有限（例如仅包含 2、3、4 等少数几种），那么可以考虑此方案。其逻辑直观：枚举所有可能的精度值，为每个精度生成一个独立的 `.round(x)` 表达式，然后利用 `pl.coalesce()` 函数，为每一行选取首个匹配（非空）的舍入结果。
sig_figs_unique = df["sig_figs_len"].unique().to_list()
df = df.with_columns(
pl.coalesce(
[
pl.when(pl.col("sig_figs_len") == x)
.then(pl.col("reverse_rate_from_euro").round(x))
for x in sig_figs_unique
]
).alias("reverse_rate_to_euro_rounded_sig_figs")
)
该方法优势在于逻辑清晰，易于调试与扩展，且 `coalesce` 保证了每行数据仅应用一条匹配的规则。然而，若精度唯一值过多（例如超过数十个），生成的表达式树会异常庞大，显著增加查询编译开销。此时，应优先采用方案一。
? 总结与选型指南
- 首选方案一（数学缩放法）：这是近乎通用的最佳实践，尤其适用于大数据集，或精度列值分布连续、范围广的场景。它实现了完全的向量化计算，无任何条件判断，性能最稳定、高效。
- 备选方案二（条件分支聚合法）：当精度位数种类极少（通常少于10种），且你需要为不同精度附加更复杂的定制逻辑（例如结合其他列进行条件判断）时，此方案可读性和灵活性更高。
- 必须规避的性能陷阱：务必避免使用 `map_rows`、`apply` 或 `iter_rows()` 等会触发 Python 层迭代的操作。它们会完全绕过 Polars 的查询优化器、并行执行与 SIMD 加速，导致性能急剧下降。
采用上述任一优化方案，最终生成的 `reverse_rate_to_euro_rounded_sig_figs` 列都将精确符合预期：例如，数值 0.154128 在 `sig_figs_len=5` 时结果为 0.15413，而 0.156006 在 `sig_figs_len=3` 时结果为 0.156。整个处理过程，都能充分保持 Polars 卓越的内存效率与闪电般的执行速度。
