python - Pandas - 检查哪一列与另一列匹配 - 不是 bool 值

Question

我在 Pandas 中有以下数据框:

target     A       B      C
0 cat      bridge  cat    brush
1 brush    dog     cat    shoe
2 bridge   cat     shoe   bridge 

期望的输出:

  target   A       B      C       D
0 cat      bridge  cat    brush   B
1 brush    dog     cat    shoe    None
2 bridge   cat     shoe   bridge  C

我需要知道哪一列与我的目标列(D 列)相匹配。我看过之前的帖子 43093394，但我只知道列是否匹配是不够的，我需要知道哪个列匹配哪个目标，如果没有匹配。同样的问题发布在这里 40320476，但是在 sql 中所以它没有帮助我。

期待您的意见，谢谢!

Answer 1

使用 eq 和 dot

df.loc[:,'A':].eq(df.target,0).dot(df.columns[1:])
Out[466]: 
0    B
1     
2    C
dtype: object
#df['D']=df.loc[:,'A':].eq(df.target,0).dot(df.columns[1:])