[問題] dataframe 欄位 多元素 判斷

作者: loser113 (洨大魯蛇ㄍ)   2020-10-07 16:08:22
找出單一元素在欄位
df[df['A']='a']
我問題是找多元素(list)在某欄位
如果不用or 寫的話怎麼寫比較好 因為list太多
q=['a','b',...]
df[df['A']='a'or df['A']='b' ....]
另外多元素再多欄位寫法
df[df['A']='a'or df['B']='a'or df['A']='b' ....]
除了for loop還有沒有更簡單寫法 感謝
作者: TitanEric (泰坦)   2020-10-07 18:34:00
先用list存boolean arrays 再用np.logical_or.reduce一次判斷還有你這個條件比較適合用df.isin可以參考一下連結的解法 shorturl.at/orvRV
作者: sherees (ShaunTheSheep)   2020-10-08 13:27:00
Isin
作者: yuasa (XD)   2020-10-08 20:59:00
isin真的好用
作者: aocaoef (aocaoef)   2020-10-09 21:53:00
want = list("a", "b", "c")df[df['A'].isin(want)]

Links booklink

Contact Us: admin [ a t ] ucptt.com