pandas利用函数或映射进行数据转换

最新推荐文章于 2024-06-27 09:32:36 发布

Johnny_Cuii

最新推荐文章于 2024-06-27 09:32:36 发布

阅读量3.5k

点赞数

分类专栏：工具类文章标签：函数数据

本文链接：https://blog.csdn.net/cuihuijun1hao/article/details/78249064

版权

本文展示了如何使用pandas DataFrame的映射功能和lambda函数将'food'列转换为对应的动物来源。通过将食物名称转换为小写并查找对应的动物，创建了新的'animal'列。

摘要由CSDN通过智能技术生成

data = DataFrame({‘food’:[‘bacon’,’pull_pork’,’bacon’,’Pastrami’,’corned beef’,’Bacon’,’pastrami’,’honey ham’,’novs lox’],
‘ounces’:[4,3,12,6,7.5,8,3,5,6]})
print(data)
meat_to_animal = {
‘bacon’:’pig’,
‘pulled pork’:’pig’,
‘pastrami’:’cow’,
‘corned beef’:’cow’,
‘honey ham’:’pig’,
‘nova lox’:’salmon’
}
print(meat_to_animal)
data[‘animal’] = data[‘food’].map(str.lower).map(meat_to_animal)
print(data)
以上用的是映射的方式，我们也可以定义一个函数

import pandas as pd
from pandas import *
import numpy as np
data = DataFrame({‘food’:[‘bacon’,’pulled pork’,’bacon’,’Pastrami’,’corned beef’,’Bacon’,’pastrami’,’honey ham’,’nova lox’],
‘ounces’:[4,3,12,6,7.5,8,3,5,6]})
print(data)
meat_to_animal = {
‘bacon’:’pig’,
‘pulled pork’:’pig’,
‘pastrami’:’cow’,
‘corned beef’:’cow’,
‘honey ham’:’pig’,
‘nova lox’:’salmon’
}
result = data[‘food’].map(lambda x:meat_to_animal[x.lower()])
print(data)