Python Polars Zip 列表列表列

Question

我有两列

pl.List(pl.List(str))

类型的列，我想按元素附加像

;

这样的分隔符。

给定一个数据框

df = pl.DataFrame({"A":[["Hi", "my", "is"]], "B":[["there", "name", "mr.x"]]})

我期望输出为

pl.DataFrame({"A":[["Hi;there", "my;name", "is;mr.x"]]})

我怎样才能在本地做到这一点？

Answer 1

(
    df
    .with_row_index()
    .explode("A","B")
    .select(A = pl.col.A + ';' + pl.col.B)
    .group_by("index")
    .agg(pl.all())
    .drop("index")
)