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

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

你可以用这个：

df.set_index('ID').lt(2).idxmin(axis=1)

ID
A    three
B      one
C      one

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

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

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