如何制作熊猫分类栏目

Question

给出一列“食物”(苹果、香蕉、胡萝卜、甜甜圈、鸡蛋……)，我想让“类别”列包含与“食物”列中的每一项相对应的值。

例如。根据下面的信息

import pandas as pd

fruit =['apple', 'banana', 'orange']
veg =['carrot', 'onion']
meat=['chicken', 'pork', 'beef']

food = fruit + veg + meat

df = pd.DataFrame(food, columns=['food'])
df

当我像这样编写代码时：

df[df['food']=='apple']['category']='fruit'
df[df['food']=='carrot']['category']='vegetable'

然而，当我以这种方式写下时，就会出现SettingWithCopyWarning。

设置此值的最佳方法是什么？

Answer 1

你可能从熊猫那里得到了SettingWithCopy警告。你可以用几种不同的方法来解决这个问题：

# Use loc
df['category'] = None # Initialize an empty column
df.loc[df['food']=='apple',  'category'] = 'fruit'
df.loc[df['food']=='carrot', 'category'] = 'vegetable'

# Use map
df['category'] = df['food'].map({
    'apple': 'fruit',
    'carrot': 'vegetable'
})