我有一个数据帧的对象，多列：business_id，categories，type_of_business...
我已成功地创建一个较小的数据帧只business_id和categories由原始数据框对象列的索引。
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对象中感兴趣。
提前致谢。
您可以使用map来做到这一点：
df[df.categories.map(lambda cats: 'Restaurants' in cats)]
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句