如何将多个参数传递给apply函数

Question

我有一个名为count的方法,它有两个参数.我需要使用apply()方法调用此方法.但是,当我将两个参数传递给apply方法时,它会给出以下错误:

TypeError:counting()只需2个参数(给定1个)

我已经看到了以下线程python pandas:将一个带有参数的函数应用于一个系列.更新,我不想使用functool.partial,因为我不想导入其他类,以便能够传递参数.

def counting(dic, strWord):
    if strWord in dic:
        return dic[strWord]
    else:
        return 0

DF['new_column'] = DF['dic_column'].apply(counting, 'word')

如果我给出一个参数,它可以工作:

def awesome_count(dic):
    if strWord in dic:
       return dic[strWord]
    else:
       return 0

DF['new_column'] = DF['dic_column'].apply(counting)

Answer 1

你可以使用lambda:

DF['new_column'] = DF['dic_column'].apply(lambda dic: counting(dic, 'word'))

另一方面,在partial这里使用绝对没有错:

from functools import partial
count_word = partial(counting, strWord='word')
DF['new_column'] = DF['dic_column'].apply(count_word)

正如@EdChum所提到的,如果您的counting方法实际上只是查找单词或将其默认为零,您可以使用方便的dict.get方法而不是自己编写一个:

DF['new_column'] = DF['dic_column'].apply(lambda dic: dic.get('word', 0))

并且lambda通过operator模块无法完成上述操作:

from operator import methodcaller
count_word = methodcaller(get, 'word', 0)
DF['new_column'] = DF['dic_column'].apply(count_word)