Polars 自定义函数指南

Polars 自定义函数指南

Polars 提供了 map 和 apply 两种表达式来处理自定义函数。

1. map 函数

map 函数将表达式所支持的 Series 数据原封不动地传递。

例子

在 groupby 中使用 map 时需要小心，因为它可能会导致不正确的结果。例如，对于一个分组的数据帧，map 会在聚合之前应用函数。

df = pl.DataFrame({
   
    "keys": ["a", "a", "b"],
    "values": [10, 7, 1],
})
out = df.groupby("keys", maintain_order=