如何在Python Polars库中调整DataFrame单个列的位置？

你可以通过操作列名列表实现需求，不用手动写出所有列的完整顺序，具体步骤如下：

1. 获取DataFrame当前的列名列表并复制
2. 移除要移动的目标列名
3. 在指定的目标索引位置插入该列名
4. 用select方法根据新的列名列表重排DataFrame

完整代码示例：

import polars as pl

# 创建示例DataFrame
data = {
    'A': [1, 2, 3],
    'B': [4, 5, 6],
    'C': [7, 8, 9],
    'D': [10, 11, 12]
}
df = pl.DataFrame(data)

# 定义要移动的列和目标索引
target_col = 'C'
target_idx = 1

# 操作列名列表调整顺序
cols = df.columns.copy()
cols.remove(target_col)
cols.insert(target_idx, target_col)

# 重排DataFrame
df_reordered = df.select(cols)

print(df_reordered)

运行后输出与你期望的一致：

shape: (3, 4)
┌─────┬─────┬─────┬──────┐
│ A   │ C   │ B   │ D    │
│ --- │ --- │ --- │ ---- │
│ i64 │ i64 │ i64 │ i64  │
╞═════╪═════╪═════╪══════╡
│ 1   │ 7   │ 4   │ 10   │
│ 2   │ 8   │ 5   │ 11   │
│ 3   │ 9   │ 6   │ 12   │
└─────┴─────┴─────┴──────┘

这种方法的优势：

• 无需硬编码所有列名，适配列数量多或列名动态变化的场景
• 严格保留其他列的原有顺序
• 支持任意目标索引（包括开头、中间、末尾位置）

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