python - 为什么我的生成器函数总是返回相同的值？

Question

我想为伯努利三角形构建一个生成器，三角形i中直线中的数字是帕斯卡三角形中j第一个数字的部分和。ij

三角形看起来像这样：

这将在 python 中由列表表示：

[[1], [1,2], [1,3,4], [1,4,7,8] 

我的生成器函数[1]作为第一个输出返回，这是正确的，但它一直返回[]！

那是我的代码：

def next_row(row):
    n = len(row)
    new_row = [row[0]] + [row[i] + row[i+1] for i in range(n - 1)] + [row[-1]]
    return new_row


def generate_pascal():
    row =[1]
    while True:
        yield row
        row=next_row(row)


def generate_bernoulli():

    row=next(generate_pascal())
    n=len(row)
    while True:
        yield row
        row=[row[0]+sum(row[0:i]) for i in range(n-1)]

Answer 1

首先，您只调用next一次，而不是每次迭代都调用它。其次，您从未在每次迭代中更新行。另外，你的部分总和yield被互换了。我通过将调用next放在 while 循环内部并在其pascal外部初始化生成器来修复这些问题。最后，你的总和有点少；我也解决了这个问题。正确的代码如下：

def generate_bernoulli():
    pascal = generate_pascal()
    while True:
        row=next(pascal)
        n=len(row)
        row=[sum(row[0:i+1]) for i in range(n)]
        yield row

Answer 2

您需要计算帕斯卡三角形各行的部分总和。这是我编写代码的方式。

def partial_sums(xs):
    s = 0
    for x in xs:
        s += x
        yield s

def generate_bernoulli():
    for row in generate_pascal():
        yield list(partial_sums(row))