pandas indexing using multiple isin clauses

后端 未结 3 885
陌清茗
陌清茗 2021-01-15 05:40

If I want to do is-in testing on multiple columns at once, I can do:

>>> from pandas import DataFrame
>>> df = DataFrame({\'A\': [1, 2, 3]         


        
3条回答
  •  暖寄归人
    2021-01-15 06:19

    You could put both the isin conditions in &

    df[df['A'].isin([1, 3]) & df['B'].isin([4, 7, 12])]
       A  B   C
    2  3  7  18
    

    You could also use query function like

    c_a = [1, 3]
    c_b = [4, 7, 12]
    df.query('(B in @c_b) & (A in @c_a)')
    
       A  B   C
    2  3  7  18
    

提交回复
热议问题