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

Question

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

TypeError: counting() takes exactly 2 arguments (1 given)

我看到了以下帖子 python pandas: apply a function with arguments to a series. Update而且我不想使用 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 提到的，如果您的计数 方法实际上只是查找一个单词或将其默认为零，您可以使用方便的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)