python - 返回满足 pandas 条件的第一列号

Question

我有一个包含多列累积和的数据集。对于每一行，我想返回满足条件的第一个列号。

玩具示例:

df = pd.DataFrame(np.array(range(20)).reshape(4,5).T).cumsum(axis=1)

>>> df
   0   1   2   3
0  0   5  15  30
1  1   7  18  34
2  2   9  21  38
3  3  11  24  42
4  4  13  27  46

例如，如果我想返回值大于20的第一列。

期望的输出:

3
3
2
2
2

一如既往地非常感谢!

Answer 1

尝试使用idxmax

df.gt(20).idxmax(1)
Out[66]: 
0    3
1    3
2    2
3    2
4    2
dtype: object