Combine pandas DataFrame query() method with isin()

后端 未结 3 1944
礼貌的吻别
礼貌的吻别 2020-12-14 07:53

So I want to use isin() method with df.query(), to select rows with id in a list: id_list. Similar question was asked bef

3条回答
  •  死守一世寂寞
    2020-12-14 08:28

    This appears to work:

    >>> df.query('a == {0}'.format(id_list))
       a  b  c  d
    0  a  a  4  1
    1  a  a  0  7
    2  b  a  2  1
    3  b  a  0  1
    4  c  b  4  0
    5  c  b  4  2
    

    Whether or not it is more clear is a matter of personal taste.

提交回复
热议问题