Python панды dataframe нарезку, с условием, если

Question

У меня есть объект DataFrame, с несколькими столбцами: business_id, categories, type_of_business ...

мне удалось создать меньший DataFrame с только business_id и categories путем индексации столбца на исходный объект DataFrame.

categories - это список определенных строк. Пример: ['Restaurant, 'food', 'bakery'] - для каждого business_id.

Одна из категорий: Restaurants. Как получить только те бизнес-идентификаторы, где слово Restaurants находится в списке категорий.

псевдокод:

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

Я заинтересован в том, как я бы включить условие if в объекте DataFrame.

Заранее спасибо.

Answer 1

Вы можете сделать это с картой:

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

Answer 2

Выбор строка в соответствии с булевым условием называется маскирования в документации.

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

Как в стороне, я вижу, что вы были опущены. Лучше, если вы включите несколько строк образца вашего DataFrame и пример желаемого результата.

Чтобы сделать его регистром нечувствительным, придерживайтесь .str.lowercase() до .isin и создавайте список категорий в нижнем регистре.