基于列表和Python字符串的组变量

Question

我的python代码没有返回我所期望的结果，我希望您能帮助我

我有一个数据集，其中包括一个城市中场馆的长列表和一列场馆类别类型（例如“意大利餐厅”）。现在，我想在我的数据框架中增加一列，其中包含更广泛的类别组（“eating”），该类别组基于一个包含字符串的列表进行搜索

我希望通过这四个示例场馆类别在我的数据框架中看到以下输出：

• “意大利餐厅”→ '吃
• "床与床,；早餐→ '睡觉的
• “现代艺术博物馆”→ '观光'
• “健身房”→ '其他的

我试着用以下方法来解决这个问题：

sleeping = ['bed','hostel','hotel']

eating = ['bar','bistro','cafe','pub','restaurant']

sightseeing = ['museum','theater','zoo']

def catgroup(cat):
    for cat in df['venue_cat']:
        if any(s in cat for s in sleeping):
            return 'sleeping'
        elif any(s in cat for s in eating):
            return 'eating'
        elif any(s in cat for s in sightseeing):
            return 'sightseeing'
        else:
            return 'other'

接

df['cat_group'] = df['venue_cat'].apply(catgroup)

不幸的是，所有场馆从第一个elif声明返回相同的类别：饮食。 我知道这是第一个elif声明，因为如果我改变elif的顺序（吃还是观光），我只会得到：观光

我很想听听你对这个问题的解决方案，因为我看不出来

Answer 1

删除函数中的for循环，以便：

def catgroup(cat):
    if any(s in cat for s in sleeping):
        return 'sleeping'
    elif any(s in cat for s in shopping):
        return 'shopping'
    elif any(s in cat for s in eating):
        return 'eating'
    elif any(s in cat for s in sightseeing):
        return 'sightseeing'
    else:
        return 'other'

df['cat_group'] = df['venue_cat'].apply(catgroup)