python - 如何将python列表附加到polars-dataframe的另一个列表（系列）？

Question

我有一个这样的极坐标数据框：

test=pl.DataFrame({"myColumn": [[1,2,3],[1,2,3],[1,2,3]]})

现在我想从另一个列表中追加列表元素，让我们说[4,5]每个条目，所以要[[1,2,3,4,5],[1,2,3,4,5],[1,2,3,4,5]]

Q1：那会怎么做？Q2：有什么方法可以让它变快？

Answer 1

dtype 的 Polars 系列/列List具有.arr（用于数组）命名空间。您可以使用该arr.concat方法附加列表。

df = pl.DataFrame({"my_column": [[1,2,3],[1,2,3],[1,2,3]]})
df.with_column(pl.col("my_column").arr.concat([4, 5]))

输出是：

shape: (3, 1)
┌───────────────┐
│ my_column     │
│ ---           │
│ list [i64]    │
╞═══════════════╡
│ [1, 2, ... 5] │
├╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌┤
│ [1, 2, ... 5] │
├╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌┤
│ [1, 2, ... 5] │
└───────────────┘