Python：如何向 itertools 函数添加参数？

我想创建一个函数，该函数返回一个创建迭代器的函数（如 itertools 中的某些内容），但具有由外部函数设置的参数。

例如，假设我想计算一个运行总和，但排除大于给定截止值（这是函数的参数）的值。

所以，假设我有一个这样的列表：

x = [1,1,25,1]

而且我想要一个函数来计算运行总和，但忽略大于 10 的值，给我这样的输出：

y = [1,2,3,4,5,6]

此代码可以解决问题：

t = 10
accumulate(x,lambda x,y:x if y>=t else x+y)

但现在我想创建一个函数，它返回一个类似上面的函数，我可以将值 t 作为参数传递给它。这是我试过的：

def sum_small_nums(t):
    def F(x):
        new_x = accumulate(x,y: x if y>=t else x+y)
    return F

不幸的是，这似乎不起作用。我可以从上面创建一个函数，但是当我尝试调用它时，我没有得到我期望的输出（例如，我得到的是“无”而不是列表）：

sum_under_10 = sum_small_nums(10)
print(sum_under_10)
x = [1,1]
x2 = sum_under_10(x)
print(x2)

<function sum_small_nums.<locals>.F at 0x7feab135a1f0>
None

你就在附近！需要从内部函数return列表：

def sum_small_nums(t):
    def F(x):
        new_x = accumulate(x,lambda x,y: x if y >= t else x + y)
        return list(new_x)
   return F

当您没有从函数中 return 任何内容时，None 会被隐式返回。

相关问答