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

Question

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

ID  TEXT  HAVE A?

1   A     YES

1   B     YES

2   A     YES

3   B     NO

Answer 1

Series.eq使用then groupbyonID和transformusing创建布尔掩码any，最后使用从该掩码或基于该掩码np.where选择值：YESNOm = df['TEXT'].eq('A').groupby(df['ID']).transform('any')df['HAS A?'] = np.where(m, 'YES', 'NO')   ID TEXT HAS A?0   1    A    YES1   1    B    YES2   2    A    YES3   3    B     NO