小能豆

pandas数据框中的自定义排序

python

我有 python pandas 数据框,其中有一列包含月份名称。

如何使用字典进行自定义排序,例如:

custom_dict = {'March':0, 'April':1, 'Dec':3}  

阅读 19

收藏
2024-07-09

共1个答案

小能豆

Pandas 0.15 引入了分类系列,它允许更清晰地实现这一点:

首先将月份列设为分类列,并指定要使用的顺序。

In [21]: df['m'] = pd.Categorical(df['m'], ["March", "April", "Dec"])

In [22]: df  # looks the same!
Out[22]:
   a  b      m
0  1  2  March
1  5  6    Dec
2  3  4  April

现在,当您对月份列进行排序时,它将根据该列表进行排序:

In [23]: df.sort_values("m")
Out[23]:
   a  b      m
0  1  2  March
2  3  4  April
1  5  6    Dec

注意:如果值不在列表中,它将被转换为 NaN。


对于那些感兴趣的人来说,这是一个较旧的答案......

您可以创建一个中间系列,然后set_index

df = pd.DataFrame([[1, 2, 'March'],[5, 6, 'Dec'],[3, 4, 'April']], columns=['a','b','m'])
s = df['m'].apply(lambda x: {'March':0, 'April':1, 'Dec':3}[x])
s.sort_values()

In [4]: df.set_index(s.index).sort()
Out[4]: 
   a  b      m
0  1  2  March
1  3  4  April
2  5  6    Dec
2024-07-09