Задача для новичков

Получилось далеко не для новичков, зато быстро работает довольно быстро.

def fib():
a, b = 0, 1
yield a
yield b
while True:
c = b
a += b
yield a
b = a
a = c

def main():
num = int(input())
gen = fib()
for i in range(num+1):
print(next(gen),end = ", ")
print()

if __name__ == '__main__':
main()

Код

def fib():

    a, b = 0, 1

    yield a

    yield b

    while True:

        c = b

        a += b

        yield a

        b = a

        a = c

       

def main():

    num = int(input())

    gen = fib()

    for i in range(num+1):

        print(next(gen),end = ", ")

    print()



if __name__ == '__main__':

    main()