查询数据框以根据条件列表/数组返回行答案

【问题标题】：Querying a dataframe to return rows based on a list/ndarray of conditions查询数据框以根据条件列表/数组返回行
【发布时间】：2022-11-25 23:39:03
【问题描述】：

假设我有一个数据框“df”：

和一组数字，称为“配置文件”：

[310, 47, 161, 51, 78, 162, 303, 314, 176, 54]

我正在尝试查询“dayNo”列上的“df”以仅返回与上面的数组（配置文件）匹配的行，但不确定如何查询。我尝试了以下，但无济于事：

df2 = df.loc[df['dayNo'] == [np.array([profiles], dtype=bool)]]

非常感谢任何帮助，谢谢！

【问题讨论】：

【解决方案1】：

您可以将布尔索引与 pandas.Series.isin 一起使用：

df2 = df.loc[df['dayNo'].isin(profiles)]

df2 = df.query('dayNo in @profiles')

【讨论】：