将函数应用于带有两个参数的pandas数据框中的列

Question

说我有一个映射:

mapping = {
    'cat': 'purrfect',
    'dog': 'too much work',
    'fish': 'meh'
    }

和a dataframe:

    animal  name       description
0   cat     sparkles   NaN
1   dog     rufus      NaN
2   fish    mr. blub   NaN

我想以编程方式description使用animal列和mappingdict作为输入填写列:

def describe_pet(animal,mapping):
    return mapping[animal]

当我尝试使用pandas apply()功能时:

df['description'].apply(describe_pet,args=(df['animal'],mapping))

我收到以下错误:

TypeError: describe_pet() takes exactly 2 arguments (3 given)

似乎使用apply()是很简单的将一个参数传递给函数.我怎么能用两个参数做到这一点？

Answer 1

您可以使用该map方法执行此操作,而无需编写函数或完全使用apply:

df['description'] = df.animal.map(mapping)