迭代器、for循环、生成器

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
l = [1, 2, 3]
# 一：下标访问方式
print(l[0])
print(l[1])
print(l[2])
print(l[3])  # 列表超出索引边界：IndexError
'''
结果
1
2
3
IndexError: list index out of range
'''

# 使用迭代器协议访问,不需要依赖索引迭代取值了
iter_l = l.__iter__() # 遵循迭代器协议，生成可迭代对象就是迭代器对象
while True:
    try:
        k = next(iter_l) # 可迭代器对象调用__iter__中的内置__next__方法
        print(k)
    except StopIteration: # 抛出StopIteration异常，并终止迭代
        break
'''
结果:
1
2
3
'''

# 二：遵循迭代器协议访问方式，迭代器允许惰性求值，只有在请求下一个元素时迭代器对象才会去生成它
x = 'Alex'
iter_x = x.__iter__()  # 得到迭代器对象，可迭代器对象即有__iter__又有__next__，但是：迭代器.__iter__()得到的仍然是迭代器本身
print(iter_x.__iter__()is iter_x) # True
# 打印字符串迭代器对象地址
print(iter_x)
# <str_iterator object at 0x10a5c2e80>
print(iter_x.__next__())  # 迭代器对象调用__iter__中的内置__next__方法,等同于next(iter_x),next()是系统内置函数
print(iter_x.__next__())  # 每次打印一个字符串的字符
print(iter_x.__next__())
print(iter_x.__next__())
print(iter_x.__next__())  # 捕捉StopIteration异常，终止迭代
'''
结果：
A
l
e
x
StopIteration
'''

1
2
3
4
5
6
7
8
9
dit = {'name': '3+2', 'age': 18, 'sex': '无'}
for i in dit:
    print(i)
'''
for循环工作原理:
1.执行in后面的对象dit.__iter__()方法，得到一个迭代器对象iter_dit
2.执行next(iter_dit),将得到的值赋值给i，然后执行循环体代码
3.依次循环，直到捕捉StopIteration异常，终止循环
'''

1
2
3
4
5
6
# 迭代是一个重复的过程，每次重复即一次迭代，并且每次迭代的结果都是下一次迭代的初始值
l = [1, 2, 3] 
num = 0
while num < len(l): # 迭代
    print(l[num])
    num += 1

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
'''
只要函数内部包含有yield关键字，那么函数名()的到的结果就是生成器，并且不会执行函数内部代码,需要生成器对象调用__next__()/next()方法，才可以执行函数内部代码
1、把函数做成迭代器
2、对比return，可以返回多次值，可以挂起/保存函数的运行状态
'''

def func():
    stay = yield '优乐美'
    print('停留的位置', stay) 
    yield '香飘飘'

# gen_res是一个生成器对象
gen_res = func()
# 打印生成器对象地址
print(gen_res)
# <generator object test at 0x10d7e8408>
print(gen_res.__next__())  # 生成器对象调用__next__()/next()方法，保留函数状态
print(next(gen_res))  # 从上一次保留函数状态地方，继续往下执行操作
print(gen_res.send('stay')) # 表达形式，yield 1i相当于return控制的是函数的返回值，stay = yield的另外一个特性，接受send传过来的值，赋值给stay，send传入的值和yield返回的值没有任何关系
'''
结果:
    优乐美
    香飘飘
'''

1
2
3
4
5
coffemilk = '优乐美'
coffemilk = 'Tea'
res = '喜之郎' if coffemilk == '优乐美' else '茶'
print(res)

1
2
3
4
5
6
7
8
list_res = ['蘑菇%s' % i for i in range(1, 11) if i <= 6] # 内存占用大，容易卡死
print(list_res)
'''
结果:
['蘑菇1', '蘑菇2', '蘑菇3', '蘑菇4', '蘑菇5', '蘑菇6']
'''
错误示例：
list_res = ['蘑菇%s' % i for i in range(1, 11) else '采蘑菇的小姑娘'] # 在列表解析中不能使用else语句，会标红

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
# 生成器表达式自动实现迭代器协议，迭代器协议对象调用__next__()/next()方法
murphy = ('土豆%s' % i for i in range(1, 11)) # 几乎不占内存 
# 打印生成器对象地址
print(murphy)
# <generator object <genexpr> at 0x104502408>
print(murphy.__next__())  # 生成器对象调用__next__()方法
print(next(murphy))  # next()内置函数，next()本质调用__next__()，
print(next(murphy))  
print(next(murphy))
print(next(murphy))
print(next(murphy))
print(next(murphy))
print(next(murphy))
print(next(murphy))
print(next(murphy))
# print(next(murphy))  # 捕捉StopIteration异常，终止迭代
'''
结果:
    土豆1
    土豆2
    土豆3
    土豆4
    土豆5
    土豆6
    土豆7
    土豆8
    土豆9
    土豆10
    StopIteration
'''

1
print(sum(i for i in range(10)))