我有一个称为计数的方法,需要2个参数。我需要使用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)
您可以使用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)
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句