1
2
3
4
5
6
7

def gen_square(n):
    for i in range(1, n+1):
        yield i**2

# 然后可以用在 in 里
for i in gen_square(10):
    print(i)