Question

我有一个名为count的方法，它接受2个参数。我需要使用apply（）方法调用此方法。但是当我将两个参数传递给apply方法时，它会给出以下错误：

TypeError：counting（）只需要2个参数（给定1个）

我见过以下帖子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所提到的，如果您的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)

Answer 2

接受的答案是完美的。教我一些关于Python的有趣的事情。但只是为了好玩，这里更准确地说是我们正在寻找的东西：

selected_words =  ['awesome', 'great', 'fantastic', 'amazing', 'love', 'horrible', 'bad', 'terrible', 'awful', 'wow', 'hate']
for this_word in selected_words:
    products[this_word] = products['word_count'].apply(lambda dic: dic.get(this_word, 0))

感谢您发布问题！

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

2 个答案: