pandas -Groupby并创建新的DataFrame? [英] Pandas - Groupby and create new DataFrame?

问题描述

这是我的情况-

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存在于Type中.

What I want to do is create a data frame of only Fruits, so I need to groupby such a way that Fruit exists in Type.

我尝试这样做:

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

我不知道这是否是这样做的方式，我在理解groupby时遇到了一些困难.如何获得仅包含Fruits的新DataFrame?

I don't know if that's the way of doing it, I'm having a little tough time understanding groupby. How do I get a new DataFrame of only Fruits?

推荐答案

您可以使用

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

这篇关于 pandas -Groupby并创建新的DataFrame?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！