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

Question

假设我有一个这样的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)

ID
A    three
B      one
C      one

Answer 2

您可以使用np.argmax来获取整数索引：

dfB = dfA['ID'].to_frame()

dfB['COLNUM'] = np.argmax(dfA.iloc[:,1:].gt(2).values, axis=1) + 1