Question

我在python中理解生成器时遇到了一些麻烦。假设我有以下简单的生成器：

def test_gen():
    yield 1
    yield 2
    yield 5
    yield 8
    yield 12

这只会产生一些数字。让我们说我想保持从发生器打印值，直到值超过10.我的第一次尝试只是

gen = test_gen()
for i in gen:
    while i > 10:
        print(i)

但这只会永远打印1。我找到了

gen = test_gen()
for i in gen:
    if i > 10:
        print(i)

按预期工作，因为它只是迭代生成器中的每个值，直到StopIteration被引发。然后我遇到了一个无限生成器的问题，例如这个生成素数的生成器：

def prime_gen():
    D = dict()
    n = 2
    while True:
        if n not in D:
            yield n
            D[n*n] = [n]
        else:
            for p in D[n]:
                D.setdefault(p + n, []).append(p)
            del D[n]
        n += 1

然后，如果我做了类似

的事情

primes = prime_gen()
for p in primes:
    if p < 100:
        print(p)

然后它会将所有内容打印到p，但随后会挂断。我认为它正在挂起，因为它试图检查primes生成的每个值，这些值将永远持续下去。迭代通过无限生成器直到满足条件的正确方法是什么？我找到的唯一方法是做

primes = prime_gen()
for p in primes:
    if p < 100:
        print(p)
    else:
        break

但我觉得有更多的pythonic方式。

Answer 1

itertools.takewhile()

$ cd /path/to/poky
$ source ./oe-init-build-env
$ bitbake <recipe>

Answer 2

你的break方法非常Pythonic，但我会考虑包装生成器：

for p in (i if i < 100 else StopIteration for i in primes):
    print(p)

迭代无限发生器直到满足条件

2 个答案: