10 python中的迭代器和生成器

1 迭代器

遵循了迭代器协议的对象为迭代器，即内部调用iter()方法转化成的对象为成迭代器。

1.1 迭代器的特征

• 具有内部方法next(),或者直接next()调用
• 迭代器是不可逆的，元素只能取一次

1.2 python中用到迭代器的举例

• 可迭代序列可以转换成迭代器

str='hello'
iter_str=str.__iter__()
print(iter_str)           #<str_iterator object at 0x000000000109D2B0>
print(type(iter_str))     #<class 'str_iterator'>
print(iter_str.__next__())   #h
print(iter_str.__next__())   #e
print(iter_str.__next__())   #l
print(iter_str.__next__())   #l
print(iter_str.__next__())   #o
print(iter_str.__next__())   #报错   StopIteration

除了字符串以外，同理，还可以将list,tuple,set类型的对象转化成迭代器

• for循环
  for循环内部实现机制：将可迭代序列调用iter方法，转化成迭代器，然后调用next()方法依次取出其中的值，遇到 StopIteration则循环结束。

l=[1,2,3,4,5,6]
for i in l:
    print(i)

#for循环等价于：
iter_l=l.__iter__()     #或者iter_l=iter(l)
while True:
    try:
        print(iter_l.__next__())
    except StopIteration:
        print("迭代完毕了，循环终止了")
        break

所有的可序类型，比如列表，字符串，都可以通过while序号拿到具体的元素，但不可序类型则只能通过for循环获取。

2 生成器

生成器本身也是迭代器；
生成器的返回值通过yield产生，yield相当于return,但多个yield可以生效;
生成器可以通过yield保存状态。
执行一次next,返回一个yield的值，且每个yield只能遍历一次。

• 三元表达式

name='hehe'
res='beautiful' if name =='hehe' else 'shasha'
print(res) 

• 三元表达式的使用示例

eggs =[]
for i in range(10):
    eggs.append('鸡蛋%s'%i)
print(eggs)   #['鸡蛋0', '鸡蛋1', '鸡蛋2', '鸡蛋3', '鸡蛋4', '鸡蛋5', '鸡蛋6', '鸡蛋7', '鸡蛋8', '鸡蛋9']

#二元表达式写法
l=['鸡蛋%s'%i for i in range(10)]
print(l)    #['鸡蛋0', '鸡蛋1', '鸡蛋2', '鸡蛋3', '鸡蛋4', '鸡蛋5', '鸡蛋6', '鸡蛋7', '鸡蛋8', '鸡蛋9']

#三元表达式写法
l2=['鸡蛋%s'%i for i in range(10) if i>5]
print(l2)    #['鸡蛋6', '鸡蛋7', '鸡蛋8', '鸡蛋9']

#没有四元表达式
l3=['鸡蛋%s'%i for i in range(10) if i>5 else "鹅蛋"]
print(l3)    #['鸡蛋6', '鸡蛋7', '鸡蛋8', '鸡蛋9']    #报错SyntaxError: invalid syntax 没有四元表达式

#生成器的三元表达式写法
l4=('鸡蛋%s'%i for i in range(10))     #把[]换成()即可
print(l4)  #<generator object <genexpr> at 0x0000000000A06938>
print(l4.__next__())   #鸡蛋0
print(l4.__next__())    #鸡蛋1
print(l4.__next__())   #鸡蛋2
print(l4.__next__())    #鸡蛋3

#函数法实现生成器
def xiadan():
    for i in range(100):
        yield '鸡蛋%s'%i

chicken=xiadan()
egg = chicken.__next__()
print(egg)

• send方法触发生成器

def test():
    print("开始啦")
    value=yield
    print('第一次',value)

t=test()
re=t.send(None)
print(re)   #None

• 迭代器触发的三种方式

def demo():
    print('begin run')
    yield 1
    print('run one time')
    yield 2
    print('run two time')
    yield 3
    print('run three time')
    last=yield
    print('run four time %s'%last)

t=demo()
print(t.__next__())   #1   方式1
print(next(t))            #2   方式2  
print(t.send('hhh'))   #None    方式3
print(t.send('hhh'))