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

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


ID  TEXT  HAVE A?

1   A     YES

1   B     YES

2   A     YES

3   B     NO


慕森卡
浏览 102回答 1
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
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Python