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

Question

我想创建一个函数，该函数返回一个创建迭代器的函数(如 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)

但是现在我想创建一个函数，它返回一个类似上面的函数，我可以将值 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

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

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)

返回

<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隐式返回。