多索引中的Pandas自定义排序行

import pandas as pd

# 创建一个具有多索引的DataFrame
data = {'A': [1, 2, 3, 4, 5],
        'B': [6, 7, 8, 9, 10]}
index = pd.MultiIndex.from_tuples([('x', 'a'), ('x', 'b'), ('y', 'a'), ('y', 'b'), ('z', 'a')], names=['first', 'second'])
df = pd.DataFrame(data, index=index)

# 自定义排序函数
def custom_sort(x):
    if x == 'a':
        return 1
    elif x == 'b':
        return 2
    else:
        return 0

# 对第二个索引级别进行自定义排序
df_sorted = df.sort_index(level='second', key=lambda x: x.map(custom_sort))

print(df_sorted)

         A   B
first second      
x     a      1   6
y     a      3   8
x     b      2   7
y     b      4   9
z     a      5  10