用Python处理Excel中的数值映射--菜鸟入门

版权声明：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")