python - 如何向 polars.DataFrame 添加一列空数组？

Question

我正在尝试将一列空列表添加到 python 中的极坐标数据框中。

我的代码

import polars as pl
a = pl.DataFrame({'a': [1, 2, 3]})
a.with_columns([pl.lit([]).alias('b')])

抛出

Traceback (most recent call last):
  File "<input>", line 1, in <module>
    a.with_columns([pl.lit([]).alias('b')])
  File "/usr/local/lib/python3.10/site-packages/polars/internals/lazy_functions.py", line 767, in lit
    return pli.wrap_expr(pylit(item, allow_object))
ValueError: could not convert value '[]' as a Literal

如何创建此列？

Answer 1

这对我有用。我用空列表 [] 编写了 pl.Series() 作为值:

import polars as pl
from polars import col

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

df = df.with_columns([
    col('a'),
    pl.Series('empty lists', [[]], dtype=pl.List),
    pl.lit(None).alias('null column'),
])
print(df) # print(df.collect()) (in case of LazyFrame)


┌─────┬─────────────┬─────────────┐
│ a   ┆ empty lists ┆ null column │
│ --- ┆ ---         ┆ ---         │
│ i64 ┆ list[f64]   ┆ bool        │
╞═════╪═════════════╪═════════════╡
│ 1   ┆ []          ┆ null        │
├╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌┤
│ 2   ┆ []          ┆ null        │
├╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌┼╌╌╌╌╌╌╌╌╌╌╌╌╌┤
│ 3   ┆ []          ┆ null        │
└─────┴─────────────┴─────────────┘