Python 如何在极坐标中的列表和标量列之间执行操作

import polars as pl

df = pl.DataFrame({"list": [[2, 2, 2], [3, 3, 3]], "scalar": [1, 2]})

shape: (2, 2)
┌───────────┬────────┐
│ list      ┆ scalar │
│ ---       ┆ ---    │
│ list[i64] ┆ i64    │
╞═══════════╪════════╡
│ [2, 2, 2] ┆ 1      │
│ [3, 3, 3] ┆ 2      │
└───────────┴────────┘

shape: (2, 3)
┌───────────┬────────┬───────────┐
│ list      ┆ scalar ┆ diff      │
│ ---       ┆ ---    ┆ ---       │
│ list[i64] ┆ i64    ┆ list[i64] │
╞═══════════╪════════╪═══════════╡
│ [2, 2, 2] ┆ 1      ┆ [1, 1, 1] │
│ [3, 3, 3] ┆ 2      ┆ [1, 1, 1] │
└───────────┴────────┴───────────┘

df = df.with_row_count().pipe(
    lambda df: df.join(
        df.explode("list")
        .with_columns(sub=pl.col("list") - pl.col("scalar"))
        .groupby("row_nr")
        .agg(pl.col("sub")),
        on="row_nr",
    )
)

Python相关问答推荐

在Python中为变量的缺失值创建虚拟值

使用GEKKO在简单DTE系统中进行一致初始化

在Python和matlab中显示不同 colored颜色 的图像

使用mySQL的SQlalchemy过滤重叠时间段

如何在类和classy-fastapi -fastapi- followup中使用FastAPI创建路由

根据二元组列表在pandas中创建新列

使用groupby Pandas的一些操作

计算每个IP的平均值

使用NeuralProphet绘制置信区间时出错

如果满足某些条件，则用另一个数据帧列中的值填充空数据帧或数组

如何根据一列的值有条件地 Select 前N组？

如何在表中添加重复的列？

字符串合并语法在哪里记录

python中的解释会在后台调用函数吗？

寻找Regex模式返回与我当前函数类似的结果

Gunicorn无法启动Flask应用，因为无法将应用解析为属性名或函数调用.'"'' "

使用字典或列表的值组合

如何将一组组合框重置回无 Select tkinter？

如何将返回引用的函数与pybind11绑定？

类型对象'；敌人'；没有属性'；损害'；