python之迭代器篇

时间:2021-09-06 00:27:23

一.迭代器

只要对象本身有_iter_()_方法,那它就是可迭代的

执行__iter__就会生成迭代器

迭代器有__next__用于获取值

__next__超出界限了会报StopIteration异常

迭代器是一次性的, 且只能一直往前迭代

获取生成器的方法:

迭代器 = 可迭代对象.__iter__()

迭代器 = iter(可迭代对象)
 
生成器获取下一个值的方法:
迭代器.__next__()
next(迭代器)
 
获取异常
d = {'a':1, 'b':2,'c':3}
i = iter(d)
while True:
try:
print(next(i))
except StopIteration:
break
 
 
 
for中的迭代器:

  for循环中in后面的就是迭代器

  实际上对于不是迭代器的for循环会自动处理成迭代器, 且在for中自动处理异常现象

d = {'a':1, 'b':2,'c':3}

for i in d:
print(i)
 
 
文件的迭代器:

  文件句柄本身就是一个迭代器

  迭代器既含有__next__也含有__iter__

常用数据类型与迭代器:

  可迭代对象:

    字符串, 列表, 元组, 字典, 集合, 文件

  迭代器:

    文件

  判断方法:

from collections import Iterable,Iterator

f=open('file')
#判断可迭代的
print(isinstance(f,Iterable))
#查看是否是迭代器
print(isinstance(f,Iterator))
 
二.生成器

生成器是在函数中, 修改return为yield

  效果:

    执行该函数之后, 并没有执行该函数体

    执行之后, 会返回一个生成器

    生成器调用next()函数之后, 会执行函数体到yield语句位置

    再调用next()函数之后, 会接着上次yield的位置继续执行函数体

    使用for循环调用生成会一次循环就是执行到一个yield 

    生成器是迭代器的一种

    生成器一样可能会产生StopIteration异常, 需要在执行next()语句中加入异常处理

def ger(max):

    for i in range(max):
yield i g = ger(2) try:
print(next(g))
print(next(g))
print(next(g))
print(next(g))
except StopIteration:
print("done")
 
 
生成器的应用
模拟tail -f的效果
import time

def tail(file):
with open(file, "r") as f:
f.seek(0, 2)
while True:
line = f.readline().strip()
if line:
yield line
else:
time.sleep(0.3)
continue g = tail('file')
for line in g:
print(line.strip())
 
模拟tail -f | grep的效果
import time

def tail(file):
with open(file, "r") as f:
f.seek(0, 2)
while True:
line = f.readline().strip()
if line:
yield line
else:
time.sleep(0.3)
continue def grep(pattern,lines):
for line in lines:
if pattern in line:
yield line g = tail('file')
g1 = grep('info', g) for line in g1:
print(line.strip())
 
 
三.协程函数

协程函数的定义:

    在生成器的基础上, 针对yield转换为赋值语句

  具体的表现形式如下

def eater(name):
print(name)
while True:
food, soup = yield
print("start to eat", food, soup)
print("done") g = eater("egon") next(g) g.send(("咖喱鸡肉盖饭","酸梅汤"))
 

协程函数的作用:

    yield 后面继续可以添加返回值

    yield前面的内容可以通过send函数来传递参数

    参数的个数和顺序要和定义的一致, 且不能有空缺