Question

我有以下示例数据集，我想按数据帧中未包含的自定义顺序对索引列进行排序。到目前为止，我还无法解决这个问题。示例：

import pandas as pd

data = {'s':[1,1,1,1], 
        'am':['cap', 'cap', 'sea', 'sea'], 
        'cat':['i', 'o', 'i', 'o'],
        'col1':[.55, .44, .33, .22],
        'col2':[.77, .66, .55, .44]}

df = pd.DataFrame(data=data)
df.set_index(['s', 'am', 'cat'], inplace=True)

Out[1]: 
           col1  col2
s am  cat            
1 cap i    0.55  0.77
      o    0.44  0.66
  sea i    0.33  0.55
      o    0.22  0.44

我想要的是以下内容：

Out[2]: 
           col1  col2
s am  cat            
1 sea i    0.33  0.55
      o    0.22  0.44
  cap i    0.55  0.77
      o    0.44  0.66

，我也可能想按'cat'的顺序来排序['o'，'i']。

Answer 1

使用sort_values和sort_index

df.sort_values(df.columns.tolist()).sort_index(level=1, ascending=False, 
                                                        sort_remaining=False)

              col1  col2
s   am   cat        
1   sea  i    0.33  0.55
         o    0.22  0.44
    cap  i    0.55  0.77
         o    0.44  0.66

将索引转换为categorical，以获取自定义顺序。

data = {'s':[1,1,1,1], 
            'am':['cap', 'cap', 'sea', 'sea'], 
            'cat':['i', 'j', 'k', 'l'],
            'col1':[.55, .44, .33, .22],
            'col2':[.77, .66, .55, .44]}

df = pd.DataFrame(data=data)
df.set_index(['s', 'am', 'cat'], inplace=True)

idx = pd.Categorical(df.index.get_level_values(2).values,
          categories=['j','i','k','l'],
          ordered=True)

df.index.set_levels(idx, level='cat', inplace=True)

df.reset_index().sort_values('cat').set_index(['s','am','cat'])

             col1   col2
s   am  cat     
1   cap  j   0.44   0.66
         i   0.55   0.77
    sea  k   0.33   0.55
         l   0.22   0.44

Answer 2

从Pandas 1.1开始，key参数为sort_values的另一种选择。

SORT_VALS = {"am": ["sea", "cap"]}

def sorter(column):
    if column.name not in SORT_VALS:
        return column
    mapper = {val: order for order, val in enumerate(SORT_VALS[column.name])}
    return column.map(mapper)

new_df = df.sort_values(by=["s", "am", "cat"], key=sorter)

#            col1  col2
# s am  cat            
# 1 sea i    0.33  0.55
#       o    0.22  0.44
#   cap i    0.55  0.77
#       o    0.44  0.66

您还可以在排序器中使用pd.Categorical并为自定义排序列返回分类的Series，这取决于您的情况可能会影响性能，但是请注意，有soon-to-be-fixed bug在pandas中使用Categorical排序可以防止多列排序。

熊猫自定义排序多级索引

2 个答案: