python-polars - 极性 : list to columns, 没有 `get`

Question

假设我有:

In [1]: df = pl.DataFrame({'a': [[1,2], [3,4]]})

In [2]: df
Out[2]:
shape: (2, 1)
┌───────────┐
│ a         │
│ ---       │
│ list[i64] │
╞═══════════╡
│ [1, 2]    │
│ [3, 4]    │
└───────────┘

我知道 'a' 的所有元素都是相同长度的列表。

我能做到:

In [10]: df.select([pl.col('a').arr.get(i).alias(f'a_{i}') for i in range(2)])
Out[10]:
shape: (2, 2)
┌─────┬─────┐
│ a_0 ┆ a_1 │
│ --- ┆ --- │
│ i64 ┆ i64 │
╞═════╪═════╡
│ 1   ┆ 2   │
│ 3   ┆ 4   │
└─────┴─────┘

但这涉及到硬编码2。

有没有办法在不硬编码2的情况下做到这一点？我可能事先不知道列表中有多少元素(我只知道它们都有相同数量的元素)

Answer 1

您可以convert to a struct和 .unnest() :

df.with_columns(pl.col("a").list.to_struct()).unnest("a")

shape: (2, 2)
┌─────────┬─────────┐
│ field_0 ┆ field_1 │
│ ---     ┆ ---     │
│ i64     ┆ i64     │
╞═════════╪═════════╡
│ 1       ┆ 2       │
│ 3       ┆ 4       │
└─────────┴─────────┘