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