Python按变量名与下标排序矩阵列的问题排查及修正

问题核心 #

需要对矩阵matriz的列按以下规则排序：

• 先按X₁、X₂…Xₙ的顺序排列（数字下标从小到大）
• 接着按Y₁、Y₂…Yₙ的顺序排列（数字下标从小到大）
• 最后放置B列

错误代码分析 #

• Code1问题：直接按列标题字符串排序，会出现X₁₀排在X₂前的情况（字符串排序逻辑与数字下标顺序不符）
• Code2问题：尝试给无数字下标的B列做类型转换，触发类型错误

正确实现代码 #

import re

# 示例矩阵（可替换为你的matriz变量）
matriz = [
    ["Y₂", "X₁", "B", "X₃", "Y₁"],
    [1, 2, 3, 4, 5],
    [6, 7, 8, 9, 10]
]

# 提取首行列标题
headers = matriz[0]

# 生成排序键的函数
def get_sort_key(header):
    # 匹配X/Y后跟数字的列标题
    pattern_match = re.match(r'([XY])(\d+)', header)
    if pattern_match:
        prefix, num_str = pattern_match.groups()
        # X组优先级0，Y组优先级1，数字下标转整数保证顺序
        group = 0 if prefix == 'X' else 1
        return (group, int(num_str))
    # B列单独设置优先级2，放在最后
    elif header == 'B':
        return (2, 0)
    # 其他未知列默认放最后（可根据需求调整）
    else:
        return (3, header)

# 获取排序后的列索引
sorted_col_indices = sorted(range(len(headers)), key=lambda idx: get_sort_key(headers[idx]))

# 重新排列矩阵列
sorted_matriz = [[row[idx] for idx in sorted_col_indices] for row in matriz]

# 打印结果（验证用）
for row in sorted_matriz:
    print(row)

代码说明 #

1. 正则匹配：仅对X/Y+数字格式的标题提取前缀和下标数字，避免B列触发转换错误
2. 排序键设计：用元组(组优先级, 下标数字)作为排序依据，元组比较会先按组优先级排序，再按下标数字从小到大排序，完美符合需求
3. 列索引排序：先确定排序后的列位置，再重新构造矩阵行，保证整个矩阵列顺序正确

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