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

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()