基于列和值列表的 Pandas 查询

Question

有没有一种方法可以根据列和列应该采用的关联值列表来自动化 Pandas 中的过滤过程？

请看这里的例子：

import pandas as pd
import seaborn as sns

df = sns.load_dataset('tips')

columns = ['sex']
values = ['Female']

df.query(@columns in @values)

columns2 = ['sex', 'smoker']
values2 = ['Male', 'Yes']
df.query(@columns2 in @values2)

当然，这行不通，但有办法吗？

关键是解决方案应该概括，因为列表的长度可能不同。

Answer 1

• 您需要符合query的要求
• 列表理解和f-strings很简单

import pandas as pd
import seaborn as sns

df = sns.load_dataset('tips')

columns = ['sex']
values = ['Female']

df.query(" and ".join([f"{c}=='{values[i]}'" for i,c in enumerate(columns)]))

columns2 = ['sex', 'smoker']
values2 = ['Male', 'Yes']
df.query(" and ".join([f"{c}=='{values2[i]}'" for i,c in enumerate(columns2)]))