python--yield生成器和return对比

Posted 2021-02-28 小陆同学

 

迭代是Python最强大的功能之一，是访问集合元素的一种方式。

迭代器是一个可以记住遍历的位置的对象。

迭代器对象从集合的第一个元素开始访问，直到所有的元素被访问完结束。迭代器只能往前不会后退。

迭代器有两个基本的方法：iter() 和 next()。

生成器是特殊的迭代器

def gen_yield():
    for i in range(1,10):
        for j in range(1,10):
            yield  i+j
            # return i+j

if __name__ == ‘__main__‘:
    aa = gen_yield()
    print(aa.__next__())
    print(aa.__next__())
    print(aa.__next__())
    print(aa.__next__())
    print(aa.__next__())
    print(aa.__next__())

yield和return都有返回值的作用，但不同的是，yield会记住当前迭代的位置，而return不会

如果还是不能很好的理解上面两者的区别，

def test_yield():
    for i in [1,2,3]:
        yield i

def test_return():
    for i in [1,2,4]:
        return i

if __name__ == ‘__main__‘:
    test_yield_obj = test_yield()
    print(‘这里测试yield‘)
    print(test_yield_obj.__next__())   # 1
    print(test_yield_obj.__next__())   # 2
    print(test_yield_obj.__next__())   # 3

    print(‘这里是测试return‘)
    print(test_return())   # 1
    print(test_return())   # 1
    print(test_return())   # 1