Pandas 在多列中的一行中查找特定值的第一次出现并返回列索引

Question

对于数据框：
df = pd.DataFrame({"A":[0,0],"B":[0,1],"C":[1,2],"D":[2,2]})

值为2或某个值时如何获取列名或列索引

并将其放在 df 的新列中，例如 df["TAG"]
df = pd.DataFrame({"A":[0,0],"B":[0,1],"C":[1,2],"D":[2,2],"TAG":[D,C]})

我试过了
df["TAG"]=np.where(df[cols]>=2,df.columns,'')

其中 [cols] 是 df 列的列表

到目前为止，我只能找到在 Pandas 中匹配值时如何查找行索引

在 excel 中，我们可以使用 MATCH(TRUE,INDEX($A:$D>=2,0),) 做一些方法并应用于多行

感谢任何帮助或提示

非常感谢您

Answer 1

试试idxmax:

>>> df['TAG'] = df.ge(2).T.idxmax()
>>> df
   A  B  C  D TAG
0  0  0  1  2   D
1  0  1  2  2   C
>>> 

Answer 2

试试这个：

search = df.ge(2).any()
# A    False
# B    False
# C     True
# D     True
# dtype: bool
df['TAG'] = search[search == True].index