Pandas：利用函数或字典进行数据转换--map

import numpy as np
import pandas as pd
from pandas import Series,DataFrame

一、利用字典进行映射

df = DataFrame({'食物':['苹果','橘子','黄瓜','番茄','五花肉'],
               '价格':[7,5,4,3,12],
               '数量':[5,8,3,4,2]})
map_dict = {
    '苹果':'水果',
    '橘子':'水果',
    '黄瓜':'蔬菜',
    '番茄':'蔬菜',
    '五花肉':'肉类'
}

df['分类'] = df['食物'].map(map_dict)
print(df)

   价格  数量   食物  分类
0   7   5   苹果  水果
1   5   8   橘子  水果
2   4   3   黄瓜  蔬菜
3   3   4   番茄  蔬菜
4  12   2  五花肉  肉类

二、利用函数进行映射

f = lambda x:map_dict[x]
df['食物'].map(f)

0    水果
1    水果
2    蔬菜
3    蔬菜
4    肉类
Name: 食物, dtype: object