python 迭代器、列表生成式、生成器

落日映苍穹つ 2021-11-15 13:14 382阅读 0赞

1.迭代器

迭代器是可以记住遍历位置的对象，迭代器对象从集合的第一个元素开始访问，访问完所以元素后停止。迭代器有两个最基本的方法：iter()、next()。iter()用于生成迭代器对象，next()用于访问下一个元素。

list1 =[1,2,3,4]
    #实例化迭代器对象
    it = iter(list1)
    
    #打印迭代器的下一个元素
    print(next(it))
    #打印迭代器的下一个元素
    print(next(it))
    #打印迭代器的下一个元素
    print(next(it))

创建一个迭代器的类，需要使用\_\_iter\_\_()、\_\_next\_\_()方法。\_\_iter\_\_()会返回一个特殊的迭代器对象，这个迭代器对象实现了\_\_next\_\_()方法并通过StopIteration 异常来标识执行完成。\_\_next\_\_()会返回下一个迭代器的对象。

#定义迭代器类
    class Test(object):
        def __iter__(self):
            self.a = 1
            return self
    
        def __next__(self):
            x = self.a
            self.a += 1
            return x
    #迭代器类进行实例化
    test_class = Test()
    #返回迭代器对象
    test = iter(test_class)
    #打印迭代器的下一个元素
    print(next(test))
    print(next(test))
    print(next(test))

二、列表生成式

#列表生成式结合for循环生成列表,生成一个列表包含0-9
    list1 = [x for x in range(10)]
    print(list1)
    
    #列表生成式结合多层for循环生成列表，生成一个列表，包含as、zx各字符串的组合
    list2 = [x+y for x in "as" for y in "zx"]
    print(list2)
    
    #列表生成式结合for循环及if语句生成列表，生成一个列表，包含10以内所有能被3整除的数字
    list3 = [x for x in range(10) if x%3 == 0]
    print(list3)

三、生成器

在python中使用yield的函数成为生成器，每次遇到yield函数时，都会暂停并保留当前所有的运行信息，返回yield函数的值，下一次执行next()时，从上次暂停的地方继续执行。使用生成器打印质数。

import sys
    #打印十个素数
    def prime_number(n):
        count = 1
        a = 2
        while True:
            if a == 2:
                yield a
                count += 1
            a += 1
            if count > n:
                return
            for i in range(2,a):
                if a%i == 0:
                    break
                elif i == a-1:
                    yield a
                    count += 1
    
    p = prime_number(10)
    while True:
        try:
            print (next(p), end=" ")
        except StopIteration:
            sys.exit()