如果列名与引用列匹配，Python将打印列名

Question

如果每行的“b”或“c”匹配（等于）“a”，我正在尝试在新列“结果”中打印列名。

这是一个例子，我有10多个列名。

我的df是这样的：

   a   b   c
a  0   1   2
b  4   4   5
c  6   7   6
d  9   9  11

预期结果：

   a   b   c   result
a  0   1   2      NaN 
b  4   4   5        b
c  6   7   6        c
d  9   9  11        b

Answer 1

在比较 b/c 和 a（带等式）时使用 idxmax。

此外，使用掩码（使用＜code＞来确保至少有一个有效匹配，因为＜code＞idxmax＜/code＞如果未找到True，将返回第一个假索引：

m = df[['b', 'c']].eq(df['a'], axis=0)

df['result'] = m.idxmax(1).where(m.any(1))

输出：

   a  b   c result
a  0  1   2    NaN
b  4  4   5      b
c  6  7   6      c
d  9  9  11      b

中间 m：

       b      c
a  False  False
b   True  False
c  False   True
d   True  False