Python基于列标题重排矩阵代码报错，如何修复并实现指定输出？

需求明确 #

需要按以下规则重排矩阵列：

• 优先排列固定列：'C'、'X'、'B'（仅保留矩阵中实际存在的列）
• 随后依次排列：
  1. 带数字后缀的X列（如X1、X2），按数字升序
  2. 带数字后缀的U列（如U1、U3），按数字升序
  3. 带数字后缀的L列（如L1、L3），按数字升序

错误原因分析 #

出现IndexError的核心原因是：直接引用了矩阵中不存在的固定列（比如原矩阵没有'X'列，但代码仍尝试将其加入列顺序列表），导致索引矩阵时访问了不存在的列。

修复后的代码（以Pandas DataFrame为例） #

import pandas as pd

# 示例矩阵数据
df = pd.DataFrame({
    'B': [1,2,3],
    'X2': [4,5,6],
    'U1': [7,8,9],
    'C': [10,11,12],
    'X1': [13,14,15],
    'L3': [16,17,18],
    'U3': [19,20,21],
    'L1': [22,23,24]
})

# 1. 筛选存在的固定顺序列
fixed_columns = ['C', 'X', 'B']
existing_fixed = [col for col in fixed_columns if col in df.columns]

# 2. 提取并排序带数字后缀的X列（排除单独的'X'）
sorted_x_num = sorted(
    [col for col in df.columns if col.startswith('X') and col != 'X'],
    key=lambda col: int(col[1:])
)

# 3. 提取并排序带数字后缀的U列
sorted_u = sorted(
    [col for col in df.columns if col.startswith('U')],
    key=lambda col: int(col[1:])
)

# 4. 提取并排序带数字后缀的L列
sorted_l = sorted(
    [col for col in df.columns if col.startswith('L')],
    key=lambda col: int(col[1:])
)

# 拼接最终列顺序并重排矩阵
new_column_order = existing_fixed + sorted_x_num + sorted_u + sorted_l
reordered_df = df[new_column_order]

# 输出重排后的列名
print(reordered_df.columns.tolist())
# 输出: ['C', 'B', 'X1', 'X2', 'U1', 'U3', 'L1', 'L3']

关键修复点 #

• 避免不存在的列引用：通过列表推导式[col for col in fixed_columns if col in df.columns]只保留矩阵中实际存在的固定列，彻底解决IndexError。
• 数字后缀排序：使用int(col[1:])作为排序键，确保带数字后缀的列按数字大小升序排列，而不是字符串字典序。
• 兼容缺失列：所有列筛选逻辑都基于矩阵现有列，即使某类列（如U列）完全不存在，也不会影响代码运行。

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