python - 使用 send 修改生成器

Question

我想通过仅更改下面的操作生成器函数来修改生成器。这是一个让我整天摸不着头脑的人力资源挑战。我从上一个关于偶数的 SO 问题中得到了使用 send 的想法。我无法使用素数检查器开始工作。该算法应打印前 10 个非素数。它目前正在打印 3。

from math import sqrt
from itertools import count, islice

def is_prime(n):
    if n < 2:
        return False

    for number in islice(count(2), int(sqrt(n) - 1)):
        if n % number == 0:
            return False

    return True

# this is the only portion of the code that can be changed  
def manipulate_generator(generator, n):
     if not is_prime(n):
         generator.send(n+1)


#all the code below cannot be changed 
def positive_integers_generator():
    n = 1
    while True:
        x = yield n
        if x is not None:
            n = x
        else:
            n += 1

k = int(input())
g = positive_integers_generator()
for _ in range(k):
    n = next(g)
    print(n)
    manipulate_generator(g, n)

标签： pythongenerator