与 Numpy 交互
Polars
表达式支持NumPy
ufuncs。 这里查看所有受支持的numpy函数的列表。
这意味着,如果一个函数不是由Polars
提供的,我们可以使用NumPy
,我们仍然可以通过NumPy
API进行快速的列操作。
实例
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 这里.