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

Question

我想创建一个函数，该函数 returns 一个创建迭代器的函数（类似于 itertools 中的东西），但具有由外部函数设置的参数。

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

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

x = [1, 1, 1, 1, 25, 1, 1]

我想要一个函数，该函数将采用运行总和，但忽略大于 10 的值，以便给我这样的输出：

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

这段代码可以解决问题：

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

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

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

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

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

returns

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

Answer 1

你很接近！需要 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。

Python: How can I add a parameter to an itertools function?