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

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/bqw18744018044/article/details/79963829
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
阅读更多 登录后自动展开
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页