python - 一个语句中的多个 isin 查询

Question

我有一个数据框 df，头部看起来像这样:

id      dept       paid
97   accounts   monthly
73       shop    weekly
20  marketing    yearly
70   analysis   monthly
55   advisory   monthly
42  marketing   monthly
76  marketing    weekly

我有兴趣查看来自部门子集的员工子集以及这些员工的子集。

例如，如果我想返回数据框的一个子集，即 marketing 和 advisory，并且其中只有那些支付 monthly 和每周

我会返回以下完整记录:

id       dept      paid
55   advisory   monthly
42  marketing   monthly
76  marketing    weekly

我尝试使用两个名为 deptQuery 和 paidQuery 的列表，它们看起来像:

deptQuery = ['marketing','advisory']

paidQuery = ['monthly','weekly']

并将其应用于:

queryResult = df[df['dept'].isin(deptQuery) and DeptTemplate['paid'].isin(paidQuery)] 

我也试过

queryResult = df[df['dept'].isin(deptQuery)] and df[df['paid'].isin(paidQuery)]

但我无法让它工作。我还可以尝试哪些其他事情？

Answer 1

将 & 用于按位 AND 并删除 df[:

queryResult = df[df['dept'].isin(deptQuery) & df['paid'].isin(paidQuery)]

print (queryResult)
   id       dept     paid
4  55   advisory  monthly
5  42  marketing  monthly
6  76  marketing   weekly

另一种解决方案 query :

queryResult = df.query('dept in @deptQuery & paid in @paidQuery')