版权声明:yangwy https://blog.csdn.net/weixin_43474731/article/details/86669343
对于许多数据集,你可能希望基于DataFrame中的数组、列或列中的数值进行一些转化
考虑上面Excel中的数据,对水果进行分类,实现如下效果
Python中代码实现如下
# 导入pandas包并重命名为pd
import pandas as pd
# 读取Excel中Sheet2中的数据
data = pd.DataFrame(pd.read_excel('test.xlsx', 'Sheet2'))
# 查看读取数据内容
print(data)
#定义水果和类别的映射
fruit_to_category = {
'葡萄': '浆果类',
'草莓': '浆果类',
'西瓜': '瓜果类',
'哈密瓜': '瓜果类',
'苹果': '仁果类',
'杏子': '核果类',
'桃子': '核果类'}
# 使用map方法进行元素转换
data['类别'] = data['水果'].map(fruit_to_category)
# 查看映射后的数据内容
print(data)
# 将映射的数据输出
data.to_excel("映射.xlsx")