python - 如何在 Python 中按组执行 proc rank(SAS 函数)？

Question

我想在 Python 中复制以下 SAS 代码。

proc rank data=a out=b ties=low;    
by id code date; 
var key_id;
ranks rank;   
run;

Answer 1

我会做这样的事情:

a = pd.DataFrame({'id': [0,1,1,2,2],
                  'code': ['a','b','b','c','c'],
                  'date':[-5,-4,-4,-2,-2],
                  'key_id':[0.05,3,5,0.001,-1]})

b =a.groupby(['id', 'code', 'date'])['key_id'].rank(ascending=True)
a['rank'] = b
a