与 Numpy 交互

Polars 表达式支持NumPy ufuncs这里查看所有受支持的numpy函数的列表。

这意味着,如果一个函数不是由Polars提供的,我们可以使用NumPy,我们仍然可以通过NumPyAPI进行快速的列操作。

实例

import polars as pl
import numpy as np

df = pl.DataFrame({"a": [1, 2, 3], "b": [4, 5, 6]})

out = df.select(
    [
        np.log(pl.all()).suffix("_log"),  # 对df所有列求对数
    ]
)
print(out)
shape: (3, 2)
┌──────────┬──────────┐
│ a_log    ┆ b_log    │
│ ---      ┆ ---      │
│ f64      ┆ f64      │
╞══════════╪══════════╡
│ 0.0      ┆ 1.386294 │
│ 0.693147 ┆ 1.609438 │
│ 1.098612 ┆ 1.791759 │
└──────────┴──────────┘

Gotcha's

阅读更多关于 gotcha's 这里.