Python3.x 将列表项的极列水平分解为新列

发布于11月08日

有没有一种两极原生的方式来水平分解一列列表？

例如，从这个开始:

df = pl.DataFrame(
    {
        "letters": ["a", "a", "b", "c"],
        "numbers": [[1, 1], [2, 3], [4, 5], [6, 7]],
    }
)

对这件事？

df = pl.DataFrame(
    {
        "letters": ["a", "a", "b", "c"],
        "numbers_a": [1, 2, 4, 6],
        "numbers_b": [1, 3, 5, 7]
    }
)

我知道.explode()%的方法，但AFAIK这只可能垂直使用

struct /取消嵌套

最简单的是先到.list.to_struct，然后是unnest

from string import ascii_lowercase

(
    df
    .with_columns(
        pl.col('numbers').list.to_struct(
            fields=lambda idx: f"numbers_{ascii_lowercase[idx]}",
            n_field_strategy='max_width'
            )
        )
    .unnest('numbers')
)

到达

from string import ascii_lowercase
max_width=df['numbers'].list.len().max()
(
    df
    .select(
        pl.exclude('numbers'), 
        *[pl.col('numbers').list.到达(x).alias(f"numbers_{ascii_lowercase[x]}") 
          for x in range(max_width)]
        )
    )

无论是哪种情况，您都会得到:

shape: (4, 3)
┌─────────┬───────────┬───────────┐
│ letters ┆ numbers_a ┆ numbers_b │
│ ---     ┆ ---       ┆ ---       │
│ str     ┆ i64       ┆ i64       │
╞═════════╪═══════════╪═══════════╡
│ a       ┆ 1         ┆ 1         │
│ a       ┆ 2         ┆ 3         │
│ b       ┆ 4         ┆ 5         │
│ c       ┆ 6         ┆ 7         │
└─────────┴───────────┴───────────┘

在前一种情况下，您可以更容易地链接方法，但第二种情况依赖于首先确定max_width，这不太灵活.第二种方法might性能更好，但我不是很确定，所以值得一试.