python 生成器

#生成器应用练习：
#编写一个生成器函数，实现生产等差数列的功能，该函数包含两个参数：数列起始值，步长”。
#调用该生成器函数，给定一个起始值，一个步长值，打印出对应等差数列的前20项。

def gennerator_difference(Start,step):
    for i in range(0,21):
        yield Start
        Start+=step

a=int(input('请输入起始值:'))      
b=int(input('请输入步长:'))
g=gennerator_difference(a,b)
for i in range(0,21):
    print(g.__next__())