Python: functools.partial返回错误

Question

我在使用functools.partial时遇到了一个问题。

我的代码：

selected_words = ['awesome', 'great', 'fantastic', 'amazing', 'love', 'horrible', 'bad', 'terrible', 'awful', 'wow', 'hate']

def awsome_count(x,i):
    if selected_words[i] in x:
        y=1
    else:
        y=0
    return y

partialfun=functools.partial(awsome_count,0)

partialfun(products[2]['word_count'])

products是SFrame。

错误：

TypeError                                 Traceback (most recent call last)
<ipython-input-108-e51348a5d1f0> in <module>()
----> 1 partialfun(products[2]['word_count'])

<ipython-input-66-9ba8c7128add> in awsome_count(x, i)
      1 def awsome_count(x,i):
----> 2     if selected_words[i] in x:
      3         y=1
      4     else:
      5         y=0

TypeError: list indices must be integers, not dict

我使用partial函数，是因为我想使用apply函数：

products['word_count'].apply(functools.partial(awsome_count,0)

Answer 1

您为partial()对象提供了一个位置参数：

functools.partial(awsome_count, 0)

该参数应用于first；向该参数添加了其他位置参数，因此您的调用：

partialfun(products[2]['word_count'])

变成：

awesome_count(0, products[2]['word_count'])

这对你的功能来说是错误的顺序。

如果要为函数的i参数应用默认值，请使用关键字参数而不是位置参数：

partialfun = functools.partial(awsome_count, i=0)

现在调用partialfun(products[2]['word_count'])变成

awsome_count(products[2]['word_count'], i=0)