如何在Polars表达式中高效实例化单元素数组/列表？

你遇到的问题其实很典型——用pl.concat_list()处理单元素列表确实有点“杀鸡用牛刀”，它的设计初衷是合并多个列成列表，单元素场景下会有不必要的性能开销。Polars其实提供了更轻量的方案，完美解决你的需求：

核心优化：用list.wrap()替代pl.concat_list() #

list.wrap()是Polars专门为“将单个值包装成单元素列表”设计的方法，底层实现做了针对性优化，性能比concat_list()好很多，代码也更简洁。

修改后的完整实现 #

先拿你的示例DataFrame来测试：

import polars as pl

df = pl.DataFrame({"key": [1, 2, 3, 4, 5], "value": ["A", "B", "C", "D", "E"]})
dimension_literal = "D"  # 你提到的固定dimensionName字面量

result_df = df.with_columns(
    pl.struct(
        pl.col(col).alias("value"),
        pl.lit(col).alias("lineItemName"),
        # 用list.wrap()把单个struct包装成单元素列表
        pl.struct(
            pl.col("key").alias("itemCode"),
            pl.lit(dimension_literal).alias("dimensionName")
        ).list.wrap().alias("dimensions"),
    ).alias(col)
    for col in df.columns
    if col != "key"
).drop("key")

print(result_df)

运行结果（完全符合你的预期） #

shape: (5, 1)
┌─────────────────────────────────────┐
│ value                               │
│ ---                                 │
│ struct[3]                           │
╞═════════════════════════════════════╡
│ {"A", "value", [{"itemCode":1, ...}]} │
├─────────────────────────────────────┤
│ {"B", "value", [{"itemCode":2, ...}]} │
├─────────────────────────────────────┤
│ {"C", "value", [{"itemCode":3, ...}]} │
├─────────────────────────────────────┤
│ {"D", "value", [{"itemCode":4, ...}]} │
├─────────────────────────────────────┤
│ {"E", "value", [{"itemCode":5, ...}]} │
└─────────────────────────────────────┘

为什么你之前的pl.lit([...])会报错？ #

Polars的pl.lit()只能接受静态字面量（比如固定的字符串、数字、甚至固定的列表字面量），但你在里面嵌套了pl.col("key")这种动态列表达式（它的值是随每行变化的），Polars目前不支持在lit里直接嵌套动态逻辑，所以会抛出TypeError: not yet implemented: Nested object types。而list.wrap()是对列进行转换，完全支持动态列表达式，刚好适配你的场景。

额外选项：用固定长度Array替代List #

如果你的业务场景更适合用固定长度的数组（而非可变长度的列表），可以用pl.array([...])来生成单元素数组：

pl.array([
    pl.struct(
        pl.col("key").alias("itemCode"),
        pl.lit(dimension_literal).alias("dimensionName")
    )
]).alias("dimensions")

Array类型在内存存储上更紧凑，查询性能也可能更好，适合长度固定的场景。

备注：内容来源于stack exchange，提问作者Vinz