使用send修改生成器

2024-04-18 21:34:39 发布

您现在位置:Python中文网/ 问答频道 /正文

我只想通过更改下面的“操纵生成器”函数来修改生成器。这是一项人力资源挑战,让我一整天都在挠头。我想到了使用send froma previous SO question on even numbers。我不能用素数检查器工作。算法应该打印前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)

Tags: thefromimportsendfalsereturnifis