如何在python pandas dataframe中搜索具有相同ID的所有值

Question

我试图搜索TEXT相同的所有值ID以查找它是否包含“A”，并添加一列来显示结果。有人可以帮我解决这个问题吗？谢谢你！ID  TEXT  HAVE A?1   A     YES1   B     YES2   A     YES3   B     NO

Answer 1

Series.eq使用then groupbyonID和transformusing创建布尔掩码any，最后使用从该掩码或基于该掩码np.where选择值：YESNO

m = df['TEXT'].eq('A').groupby(df['ID']).transform('any')

df['HAS A?'] = np.where(m, 'YES', 'NO')

   ID TEXT HAS A?

0   1    A    YES

1   1    B    YES

2   2    A    YES

3   3    B     NO