yield 使用理解。

类似生成一个列表。
def test(lenth):
    for i in range(lenth):
        yield i

if __name__ == '__main__':
    for x in test(5):
        print(x,end= ' ')

 

输出：

0 1 2 3 4