使用一列中的多个值过滤数据框中的值

时间:2019-08-29 03:30:20

标签: python pandas

对于以下数据框,我想根据计数值过滤数据。我想保留10、5和2:

>>>>>

我知道我可以将代码编写为

index      count
    1           10
    2           5
    3           2
    4           1
    5           6
    6           7
    7           "tab"

有没有更简单的方法?我有20多个值。我尝试了以下操作,但没有成功:

df[(df.count==10) | (df.count==5) | (df.count==2) |  (df.count=="tab")]

谢谢。

1 个答案:

答案 0 :(得分:1)

使用isin

df[df['count'].isin([10, 5, 2, "tab"])]