基于列表对pandas数据帧进行排序

时间:2014-11-03 03:03:01

标签: python pandas

我想对以下数据框进行排序:

Region           LSE          North      South
0                   Cn     33.330367   9.178917
1               Develd     -36.157025 -27.669988
2               Wetnds    -38.480206 -46.089908
3                Oands    -47.986764 -32.324991
4               Otherg    323.209834  28.486310
5                 Soys      34.936147   4.072872
6                  Wht     0.983977 -14.972555

我想对它进行排序,以便根据列表重新排序LSE列:

lst = ['Oands','Wetnds','Develd','Cn','Soys','Otherg','Wht']

当然其他列也需要相应地重新排序。在熊猫中有没有办法做到这一点?

1 个答案:

答案 0 :(得分:13)

pandas 0.15版本对Categorical s的改进支持使您可以轻松完成此操作:

df['LSE_cat'] = pd.Categorical(
    df['LSE'], 
    categories=['Oands','Wetnds','Develd','Cn','Soys','Otherg','Wht'], 
    ordered=True
)
df.sort('LSE_cat')
Out[5]: 
   Region     LSE       North      South LSE_cat
3       3   Oands  -47.986764 -32.324991   Oands
2       2  Wetnds  -38.480206 -46.089908  Wetnds
1       1  Develd  -36.157025 -27.669988  Develd
0       0      Cn   33.330367   9.178917      Cn
5       5    Soys   34.936147   4.072872    Soys
4       4  Otherg  323.209834  28.486310  Otherg
6       6     Wht    0.983977 -14.972555     Wht

如果这只是一个临时排序,那么将LSE列保持为 Categorical可能不是你想要的,但如果这个顺序是 你希望能够使用几次的东西 在不同的情况下,Categoricals是一个很好的解决方案。


pandas的更高版本中,sort已替换为sort_values,因此您需要:

df.sort_values('LSE_cat')