python - Pandas - Groupby 并创建新的 DataFrame？

Question

这是我的情况-

In[1]: data
Out[1]: 
     Item                    Type
0  Orange           Edible, Fruit
1  Banana           Edible, Fruit
2  Tomato       Edible, Vegetable
3  Laptop  Non Edible, Electronic

In[2]: type(data)
Out[2]: pandas.core.frame.DataFrame

我想做的是创建一个只有 Fruits 的数据框，所以我需要 groupby 这样 Fruit 存在于类型。

我试过这样做:

grouped = data.groupby(lambda x: "Fruit"in x, axis=1)

我不知道这是否是这样做的方式，我在理解 groupby 时遇到了一些困难。我如何获得只有 Fruits 的新 DataFrame？

Answer 1

你可以使用

data[data['Type'].str.contains('Fruit')]

---

import pandas as pd

data = pd.DataFrame({'Item':['Orange', 'Banana', 'Tomato', 'Laptop'],
                     'Type':['Edible, Fruit', 'Edible, Fruit', 'Edible, Vegetable', 'Non Edible, Electronic']})
print(data[data['Type'].str.contains('Fruit')])

产量

     Item           Type
0  Orange  Edible, Fruit
1  Banana  Edible, Fruit