python - 如何在 Pandas 中添加排序功能？

Question

我想获取列值的排序顺序并作为新列 a_order .例如，值的顺序是 0.5 < 2.5 < 3.6在 a 列中.

因此在 a_order 列中, 0.5对应0 , 2.5对应1 , 和 3.6对应2 .

Input:

import pandas as pd 
import numpy as np 
df = pd.DataFrame({'a':[0.5, 3.6, 2.5]})

Output:

     a
0  0.5
1  3.6
2  2.5

Expected:

     a  a_order
0  0.5        0
1  3.6        2
2  2.5        1

Answer 1

使用rank , 减去 1 并转换为整数:

df['a_order'] = df['a'].rank(method='dense').sub(1).astype(int)
print (df)
     a  a_order
0  0.5        0
1  3.6        2
2  2.5        1