pandas的is in和not in

简介

pandas按条件筛选数据时，除了使用query()方法，还可以使用isin和对isin取反进行条件筛选.

代码

    import pandas as pd
    df = pd.DataFrame({'a':[1, 2, 3, 4, 5, 6],
                       'b':[1, 2, 3, 4, 5, 6],
                       'c':[1, 2, 3, 4, 5, 6]})
    
    filter_condition = {'a':[1, 2, 3]}
    
    df_in = df[df.isin(filter_condition)['a']]
    df_notin = df[~df.isin(filter_condition)['a']]
df.isin(filter_condition)
       a      b      c
0   True   True  False
1   True   True  False
2   True  False  False
3  False  False  False
4  False  False  False
5  False  False  False

使用总结

pandas使用isin时，返回DataFrame中存储的数据为是否满足筛选条件的boolean，可以通过boolean对DataFrame对数据进行筛选。