如何在指定索引位置插入polars.lit生成的列？

要实现这个需求，核心问题在于insert_at_idx要求第二个参数为Series类型，而pl.lit()返回的是表达式（Expr），无法直接传入。可以通过以下两种方法解决：

方法1：将pl.lit转换为Series后插入 #

先把pl.lit()生成的常量表达式转换成与DataFrame行数匹配的Series，再调用insert_at_idx：

import polars as pl

# 示例DataFrame
dt = pl.DataFrame({"col1": [1, 2, 3], "col2": [4, 5, 6]})
insert_index = 1  # 目标插入位置

# 将pl.lit转为对应长度的Series
constant_series = pl.lit("some_text").repeat(len(dt)).alias("new_col")
# 插入到指定索引位置
result_df = dt.insert_at_idx(insert_index, constant_series)

print(result_df)

方法2：先添加列再调整列顺序（Polars推荐写法） #

利用Polars的链式操作，先通过with_columns添加常量列，再用select重新排列列的位置，达到插入到指定索引的效果：

import polars as pl

dt = pl.DataFrame({"col1": [1, 2, 3], "col2": [4, 5, 6]})
insert_index = 1
new_col_name = "new_col"

# 链式添加并调整列顺序
result_df = dt.with_columns(pl.lit("some_text").alias(new_col_name)) \
              .select(dt.columns[:insert_index] + [new_col_name] + dt.columns[insert_index:])

print(result_df)

两种方法都能实现需求：

• 方法1更贴合"指定索引插入"的直观逻辑，适合单独操作场景；
• 方法2是Polars的惯用链式写法，无需手动处理Series长度，更适合复杂数据流水线场景。

内容的提问来源于stack exchange，提问作者Sudoh