如何获取满足条件的第一列的索引

Question

假设我有这样的 A df：

ID | one | two | three  

A  | 1   | 0   |   3  

B  | 3   | 1   |   4  

C  | 4   | 3   |   2  

如何获取值大于 2 的第一列的索引？因此 O 可以得到这种关系：

ID | COLNUM  

A  |  3    

B  |  1   

C  |  1

Answer 1

你可以使用这个：df.set_index('ID').lt(2).idxmin(axis=1)IDA    threeB      oneC      one

Answer 2

您可以使用np.argmax获取整数索引：dfB = dfA['ID'].to_frame()dfB['COLNUM'] = np.argmax(dfA.iloc[:,1:].gt(2).values, axis=1) + 1