Python 生成器令人困惑的行为

Question

当我阅读生成器教程https://realpython.com/introduction-to-python-generators/时，就会出现这个问题。 is_palindrome(101)返回True，但是代码并没有打印101。我调试了一下，看到yield 101导致执行回到main for循环，然后立即返回到generator，非常混乱。 生成器产生 101，因此它应该是for j in pal_gen:中for j in pal_gen:一个元素，不是吗？

def is_palindrome(num):
    # Skip single-digit inputs
    if num // 10 == 0:
        return False
    temp = num
    reversed_num = 0

    while temp != 0:
        reversed_num = (reversed_num * 10) + (temp % 10)
        temp = temp // 10

    if num == reversed_num:
        return True
    else:
        return False

def infinite_palindromes():
    num = 0
    while True:
        if is_palindrome(num):
            i = (yield num)
            if i is not None:
                num = i
        num += 1

if __name__=='__main__':        
    pal_gen = infinite_palindromes()
    for j in pal_gen:
        print(f'j={j}')
        digits = len(str(j))
        if digits == 5:
            pal_gen.close()        
        pal_gen.send(10 ** (digits))

目前的输出是：

j=11
j=111
j=1111
j=10101
Traceback (most recent call last):
  File "generator.py", line 33, in <module>
    pal_gen.send(10 ** (digits))
StopIteration

Answer 1

你会注意到它在每个之间跳过一个。 101不见了，1001不见了。 10001 丢失。 我很确定send会导致生成器运行到下一个 yield 语句。 例如，使用以下代码：

if __name__=='__main__':
    pal_gen = infinite_palindromes()
    for j in pal_gen:
        print('j={}'.format(j))
        digits = len(str(j))
        if digits == 5:
            pal_gen.close()
        print(pal_gen.send(10 ** (digits)))

这将打印出“j”之间的缺失值：

j=11
101
j=111
1001
j=1111
10001
j=10101

考虑用一段while重做你的for语句：

if __name__=='__main__':
    pal_gen = infinite_palindromes()
    j=next(pal_gen)
    digits=0    
    while digits<5:  
        print('j={}'.format(j))
        digits = len(str(j))
        j = pal_gen.send(10 ** (digits))
    else:
        pal_gen.close()