Pandas多列排序时仅对指定列应用key参数的实现方法

没问题，这个需求完全可以实现！咱们来一步步搞定它：

核心思路 #

当你需要对多列排序，但仅给其中一列设置自定义排序规则时，关键是要给sort_values的key参数传入能分别处理各列排序依据的逻辑——针对A列提取数字部分作为排序键，B列直接用原始值按常规规则排序。

方法一：直接使用key参数的多列处理（推荐） #

在pandas 1.4.0及以上版本中，当by参数指定多个列时，key函数会接收一个包含这些列的子DataFrame。我们可以返回一个元组，其中每个元素对应by列表里列的排序依据：

import pandas as pd

df = pd.DataFrame({'A':['y2','x3','z1','z1'],'B':['y2','x3','a2','z1']})

# 仅对A列应用自定义key，B列常规排序
sorted_df = df.sort_values(
    by=['A', 'B'],
    key=lambda x: (x['A'].str[1], x['B'])
)

print(sorted_df)

执行后得到的结果完全符合你的预期：

方法二：借助临时列（兼容旧版本pandas） #

如果你的pandas版本较低，不支持上述key参数的多列处理方式，可以先创建一个临时列存储A列的数字部分，排序完成后再删除这个临时列：

import pandas as pd

df = pd.DataFrame({'A':['y2','x3','z1','z1'],'B':['y2','x3','a2','z1']})

# 添加临时列存储A列的数字部分
df['A_sort_key'] = df['A'].str[1]
# 按临时列和B列排序
sorted_df = df.sort_values(by=['A_sort_key', 'B'])
# 删除临时列
sorted_df = sorted_df.drop('A_sort_key', axis=1)

print(sorted_df)

这个方法逻辑更直观，同样能得到你想要的结果。

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