得分栏与其他栏

Question

我想对df中有多少其他列大于或等于一个参考列进行排序。给定testdf：

testdf = pd.DataFrame({'RefCol': [10, 20, 30, 40], 
                      'Col1': [11, 19, 29, 40], 
                      'Col2': [12, 21, 28, 39], 
                      'Col3': [13, 22, 31, 38] 
                      })

我正在使用helper函数：

def sorter(row):
    sortedrow = row.sort_values()
    return sortedrow.index.get_loc('RefCol')

作为：

testdf['Score'] = testdf.apply(sorter, axis=1)

用实际数据这种方法速度很慢，如何加速呢？谢谢

Answer 1

看起来您需要比较RefCol并检查是否有任何列小于RefCol，请使用：

testdf.lt(testdf['RefCol'],axis=0).sum(1)

---

0    0
1    1
2    2
3    2

大于等于使用：

testdf.drop('RefCol',1).ge(testdf.RefCol,axis=0).sum(1)