草庐IT

rankdata

全部标签

python - numpy 的 argsort 可以给相等的元素相同的等级吗?

我想得到每个元素的排名,所以我在numpy中使用argsort:np.argsort(np.array((1,1,1,2,2,3,3,3,3)))array([0,1,2,3,4,5,6,7,8])它给同一个元素不同的等级,我能得到相同的等级吗:array([0,0,0,3,3,5,5,5,5]) 最佳答案 如果你不介意对scipy的依赖,你可以使用scipy.stats.rankdata,method='min':In[14]:aOut[14]:array([1,1,1,2,2,3,3,3,3])In[15]:fromscipy.