Pandas多列组合排序后删临时列报错及索引排序问题

一、正确删除临时列vals #

你遇到的KeyError是因为df.drop('vals')默认在**行索引（axis=0）**中查找待删除对象，而vals是列，只需明确指定操作轴即可解决：

• 方法1：指定轴参数

df = df.drop('vals', axis=1)

• 方法2：用columns参数更直观

df = df.drop(columns='vals')

• 方法3：直接删除列（快速操作）

del df['vals']

二、基于元组索引的自定义排序 #

sort_index的key函数仅按索引层级单独处理，无法直接获取完整索引元组，可通过以下两种方案实现自定义排序：

方案1：索引转列排序后还原 #

将多级索引拆为临时列，用自定义逻辑排序后重新设为索引：

# 重置索引，将元组索引转为列
df_reset = df.reset_index()
# 自定义排序逻辑（示例：按索引元组第二个元素倒序、第一个元素正序）
df_sorted = df_reset.sort_values(
    by=['level_0', 'level_1'],
    key=lambda x: (-x if x.name == 'level_1' else x)
)
# 还原索引
df = df_sorted.set_index(['level_0', 'level_1'])

方案2：生成排序后索引列表再重排 #

直接提取所有索引元组，按自定义规则排序后用reindex重新排列：

# 获取所有索引元组
index_list = list(df.index)
# 自定义排序逻辑（示例：按元组长度、首元素排序）
index_list.sort(key=lambda t: (len(t), t[0]))
# 重新索引
df = df.reindex(index_list)

如果是MultiIndex，可直接操作索引元组数组：

# 按自定义规则排序索引元组
sorted_tuples = sorted(df.index.values, key=lambda t: (t[1], t[0]))
# 转为MultiIndex后重排
df = df.reindex(pd.MultiIndex.from_tuples(sorted_tuples))

内容的提问来源于stack exchange，提问作者Jędrzej Dudkiewicz