python - 如何根据特定顺序对 pandas 列进行排序？

Question

我有一个看起来像这样的 df:

index  life_stage
1      Early Childhood
2      Birth
3      Infancy
...

life_stage 列的排序不正确，我不能依赖字母顺序。

正确的顺序是

Birth
Infancy
Early Childhood

是否可以根据我在 Pandas 中指定的顺序对生命阶段列进行排序？

Answer 1

让我们使用 pd.Categorical 将“life_stage”转换为有序分类列:

df['life_stage'] = pd.Categorical(
    df['life_stage'], 
    categories=['Birth', 'Infancy', 'Early Childhood'], 
    ordered=True
)

请注意我为 pd.Categorical 指定类别的顺序。现在，使用 life_stage 调用 sort_values:

df.sort_values(by=['life_stage'])

   index       life_stage
2      1            Birth
1      2          Infancy
0      3  Early Childhood

<小时/>

仅供引用，按字母顺序对“life_stage”进行排序即可

   index       life_stage
2      1            Birth
0      3  Early Childhood   # wrong!
1      2          Infancy