主页 / coding / 问题 / 79142186
Accepted
bzm3r
Asked: 2024-10-31 01:25:59 +0800 CST

如何展平列表类型列表的列元素，使其成为具有列表类型元素的列？

  • 772

请考虑以下示例：

import polars as pl

pl.DataFrame(pl.Series("x", ["1, 0", "2,3", "5 4"])).with_columns(
    pl.col("x").str.split(",").list.eval(pl.element().str.split(" "))
)

shape: (3, 1)
┌────────────────────┐
│ x                  │
│ ---                │
│ list[list[str]]    │
╞════════════════════╡
│ [["1"], ["", "0"]] │
│ [["2"], ["3"]]     │
│ [["5", "4"]]       │
└────────────────────┘

我想将列中的元素展平，这样这些元素就不再是嵌套列表，而是列表。我该怎么做？

python-polars

1 个回答

  1. Best Answer

    您可以使用它list.explode()来为每个列表元素返回一行，并在其中使用它list.eval()来扩展每行的嵌套列表，而不是爆炸系列本身。

    import polars as pl

df = pl.DataFrame(pl.Series("x", ["1, 0", "2,3", "5 4"]))
print(df.with_columns(
    pl.col("x")
    .str.split(",")
    .list.eval(pl.element().str.split(" "))
    .list.eval(pl.element().explode())
))
# You can also combine it with your existing .list.eval()
print(df.with_columns(
    pl.col("x")
    .str.split(",")
    .list.eval(pl.element().str.split(" ").explode())
))
    • 3

相关问题