Python Polars：通过计算列表中最大值和最小值的差值，从另一个整型列创建一个整型列

发布于02月01日

我有一个像这样的polars个格子:

shape: (10, 2)
┌────────┬───────┐
│ foo    ┆ bar   │
│ ---    ┆ ---   │
│ i64    ┆ i64   │
╞════════╪═══════╡
│ 86     ┆ 11592 │
│ 109    ┆ 2765  │
│ 109    ┆ 4228  │
│ 153    ┆ 4214  │
│ 153    ┆ 7217  │
│ 153    ┆ 11095 │
│ 160    ┆ 1134  │
│ 222    ┆ 5509  │
│ 225    ┆ 10150 │
│ 239    ┆ 4151  │
└────────┴───────┘

和一个由points个整数组成的sorted列表:

points = [0, 1500, 3000, 4500, 6000, 7500, 9000, 10500, 12000]

我想创建一个新的列baz，这样对于bar中的每个元素y，我找到points中最大的x，使得x =< y.那么baz的元素是y - x.我怎么能做到这一点？

推荐答案

通过使用join_asof，您可以得到一个完全依赖极点表达式的解决方案.首先，在列表中创建DataFrame，然后连接(我假设points列表是预先排序的)

这里，我们将连接的strategy参数留空，因此将使用默认的backward策略.

backward搜索 Select 右DataFrame (in our case - list of points)中on键小于或等于左DataFrame (in our case - list of points)键的最后一行:

df_points = pl.DataFrame({'point': points})

(
    df.sort('bar')
    .join_asof(df_points.set_sorted('point'), right_on='point', left_on='bar')
    .with_columns(baz=pl.col('bar') - pl.col('point'))
    .drop('point')
    .sort('foo')
)

┌─────┬───────┬──────┐
│ foo ┆ bar   ┆ baz  │
│ --- ┆ ---   ┆ ---  │
│ i64 ┆ i64   ┆ i64  │
╞═════╪═══════╪══════╡
│ 86  ┆ 11592 ┆ 1092 │
│ 109 ┆ 2765  ┆ 1265 │
│ 109 ┆ 4228  ┆ 1228 │
│ 153 ┆ 4214  ┆ 1214 │
│ 153 ┆ 7217  ┆ 1217 │
│ 153 ┆ 11095 ┆ 595  │
│ 160 ┆ 1134  ┆ 1134 │
│ 222 ┆ 5509  ┆ 1009 │
│ 225 ┆ 10150 ┆ 1150 │
│ 239 ┆ 4151  ┆ 1151 │
└─────┴───────┴──────┘