生成器上的 Python 聚合

Question

我有一个生成器，它在每次迭代中返回一个列表。列表的每个元素可以是 0 或 1。我想计算返回的元素总数（包括 0 和 1）和返回的 1 的总数。我尝试使用这样的reduce函数来实现它：

t = reduce((lambda x,y:(y[0]+1,y[1]+x)), gen_fn(), (0,0)) 上面的

gen_fn() 是在每个 yield 语句中返回部分列表的生成器。我想通过使用元组（0,0）进行计数来实现它。鉴于从生成器返回的元素如下：

[0, 1, 1, 0, 1]

我对 t 的预期输出是 (5,3)。但是我的代码失败并显示此错误消息：

类型错误：+ 不支持的操作数类型：“int”和“tuple”

谁能帮我找出问题所在？我缺乏使用 reduce 和 lambda 函数的经验，这让我无法弄清楚我做错了什么。提前致谢。

Answer 1

我认为这里最好的答案是保持简单：

count = 0
total = 0
for item in gen_fn():
    count += 1
    total += item

在此处使用 reduce() 只会降低代码的可读性。

如果您的问题是代码高尔夫，并且您想要一个单线（同时保持惰性评估），那么您想要：

count, total = collections.deque(zip(itertools.count(1), itertools.accumulate(gen_fn())), maxlen=1).pop()

当然，如果选择这样的结构而不是简单的解决方案，你会很生气。

编辑：

如果生成器产生多个较小的部分，则只需使用itertools.chain.from_iterable(gen_fn()) 将其展平即可。

Answer 2

您的lambda 参数错误；第一个参数 (x) 是到目前为止的总数（元组），第二个参数 (y) 是新值（整数）。试试：

t = reduce((lambda x, y: (x[0]+1, x[1]+y)), gen_fn(), (0,0))

使用虚拟函数：

def gen_fn():
    for x in [0, 1, 1, 0, 1]:
        yield x

我收到(5, 3)。

来自the docs 的reduce 的等效实现可能会让事情更清楚：

def reduce(function, iterable, initializer=None):
    it = iter(iterable)
    if initializer is None:
        try:
            initializer = next(it)
        except StopIteration:
            raise TypeError('reduce() of empty sequence with no initial value')
    accum_value = initializer
    for x in it:
        accum_value = function(accum_value, x) # note value so far is first arg
    return accum_value

Answer 3

正如 jonrsharpe 所指出的，鉴于 reduce 的工作方式，您正在向后使用 lambda 参数。但是，如果从生成器生成的每个项目都是一个列表，那么如何添加内容可能会存在进一步的问题。

这个问题是你的y 值（生成器产生的项目）不是一个数字，而是一个列表。您需要计算它的长度和 1s 的数量，因此您可能希望您的 lambda 函数为：

lambda x, y: (x[0]+len(y), x[1]+sum(y))

Answer 4

采取完全不同的方法怎么样？

t = [(len(row), len(filter(lambda x: x == 1, row))) for row in gen_fn()]