python - 按特定顺序对 Pandas DataFrame 中的列进行排序

Question

给定这个 DataFrame:

df = pd.DataFrame([['August', 2], ['July', 3], ['Sept', 6]], columns=['A', 'B'])

我想按以下顺序对 A 列进行排序:7 月、8 月、9 月。是否有某种方法可以使用像“sort_values”这样的排序函数，但按值预先定义排序顺序？

Answer 1

使用分类

df.A=pd.Categorical(df.A,categories=['July', 'August', 'Sept'])
df=df.sort_values('A')
df
Out[310]: 
        A  B
1    July  3
0  August  2
2    Sept  6