我試圖搜索TEXT相同的所有值ID以查找它是否包含“A”,并添加一列來顯示結果。有人可以幫我解決這個問題嗎?謝謝你!ID TEXT HAVE A?1 A YES1 B YES2 A YES3 B NO
1 回答

達令說
TA貢獻1821條經驗 獲得超6個贊
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
添加回答
舉報
0/150
提交
取消