将pandas DataFrame query（）方法与isin（）结合使用

将pandas DataFrame query（）方法与isin（）结合使用

从文档的

query

您可以在环境中引用变量，方法是在变量前添加’@’字符，例如

@a + b

。

在您的情况下：

In [38]: df.query('a == @id_list')Out[38]:   a  b  c  d0  a  a  3  41  a  a  4  52  b  a  2  33  b  a  1  54  c  b  2  45  c  b  1  2