Python pandas数据帧切片，带有if条件答案

【问题标题】：Python pandas dataframe slicing, with if conditionPython pandas数据帧切片，带有if条件
【发布时间】：2013-09-13 15:39:29
【问题描述】：

我有一个 DataFrame 对象，有多个列：business_id、categories、type_of_business...

我已经设法通过在原始 DataFrame 对象上的列索引创建了一个只有 business_id 和 categories 的较小 DataFrame。

categories 是特定字符串的列表。例如：['Restaurant, 'food', 'bakery'] - 对应每个business_id。

其中一个类别是Restaurants。我将如何仅检索类别列表中包含单词 Restaurants 的那些企业 ID。

伪代码：

for row in smaller_DataFrame:
    if 'Restaurants' in row['categories']:
        add this business_id to some dictionary.

我对如何将 if 条件合并到 DataFrame 对象中感兴趣。

提前致谢。

【问题讨论】：

【解决方案1】：

根据布尔条件选择行在文档中称为屏蔽。

df[df['categories'].isin(['Restaurant', 'food', 'bakery'])]

顺便说一句，我看到你被否决了。最好包含一些 DataFrame 示例行和所需结果的示例。

要使其不区分大小写，请将.str.lowercase() 放在.isin 之前，并将类别列表全部小写。

【讨论】：

【解决方案2】：

你可以用地图做到这一点：

df[df.categories.map(lambda cats: 'Restaurants' in cats)]

【讨论】：