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

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)

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