如何在一个条件下子集数据帧

Question

我有这样一个熊猫数据帧：

 buyer_id meal_type
    139       Veg
    140    NonVeg
    140       Veg
     36    NonVeg
     79       Veg
     79    NonVeg
     79    NonVeg
     72    NonVeg
     72    NonVeg
     65    NonVeg
     65       Veg

现在我想要所有的buyer_id只有Veg作为膳食类型，所有的buyer_id只有Veg and NonVeg作为膳食类型，所有的buyer_id只有NonVeg作为膳食类型

所以

139 Veg
140 Veg and NonVeg
36  NonVeg
79  Veg and NonVeg

等等。我在Python中执行以下操作来获取所有具有Veg、NonVeg或两者兼有的买家id

segments_data.buyer_id[segments_data['meal_type']=='Veg' &      
segments_data['meal_type']=='NonVeg']

但它不起作用。请帮忙

Answer 1

按buyer_id分组并将meal_type转换为一个集合：

>>> df.groupby('buyer_id')['meal_type'].apply(set).str.join(' and ')

    buyer_id
36             NonVeg
65     Veg and NonVeg
72             NonVeg
79     Veg and NonVeg
139               Veg
140    Veg and NonVeg
Name: meal_type, dtype: object