这就是我的情况——
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.
我尝试这样做：
grouped = data.groupby(lambda x: "Fruit" in x, axis=1)
我不知道这是否是这样做的方式，我有点难以理解groupby。我如何获得新的DataFrame仅有的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')])
yields
Item Type
0 Orange Edible, Fruit
1 Banana Edible, Fruit
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
