我試圖搜索TEXT相同的所有值ID以查找它是否包含“A”，并添加一列來顯示結(jié)果。有人可以幫我解決這個問題嗎？謝謝你！ID TEXT HAVE A?1 A YES1 B YES2 A YES3 B NO
1 回答
達(dá)令說
TA貢獻(xiàn)1821條經(jīng)驗 獲得超6個贊
Series.eq使用then groupbyonID和transformusing創(chuàng)建布爾掩碼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
提交
取消