[问题] 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