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

来源:互联网 发布:借壳上市 知乎 编辑:程序博客网 时间:2024/06/03 04:54

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)

阅读全文
0 0
原创粉丝点击