Python学习笔记六:迭代与生成

1、迭代

如果给定一个list或tuple,我们可以通过for循环来遍历这个list或tuple,这种遍历我们称为迭代(Iteration)。

在Python中,迭代是通过for … in来完成的

只要是可迭代对象,无论有无下标,都可以迭代,比如dict就可以迭代:

>>> d = {'a': 1, 'b': 2, 'c': 3}
>>> for key in d:
...     print(key)
...
a
c
b

因为dict的存储不是按照list的方式顺序排列,所以,迭代出的结果顺序很可能不一样。
默认情况下,dict迭代的是key。如果要迭代value,可以用for value in d.values(),如果要同时迭代key和value,可以用for k, v in d.items()。

由于字符串也是可迭代对象,因此,也可以作用于for循环:

>>> for ch in 'ABC':
...     print(ch)
...
A
B
C

所以,当我们使用for循环时,只要作用于一个可迭代对象,for循环就可以正常运行,而我们不太关心该对象究竟是list还是其他数据类型。

那么,如何判断一个对象是可迭代对象呢?方法是通过collections模块的Iterable类型判断:

>>> from collections import Iterable
>>> isinstance('abc', Iterable) # str是否可迭代
True
>>> isinstance([1,2,3], Iterable) # list是否可迭代
True
>>> isinstance(123, Iterable) # 整数是否可迭代
False

最后一个小问题,如果要对list实现类似Java那样的下标循环怎么办?Python内置的enumerate函数可以把一个list变成索引-元素对,这样就可以在for循环中同时迭代索引和元素本身:

>>> for i, value in enumerate(['A', 'B', 'C']):
...     print(i, value)
...
0 A
1 B
2 C

上面的for循环里,同时引用了两个变量,在Python里是很常见的,比如下面的代码:

>>> for x, y in [(1, 1), (2, 4), (3, 9)]:
...     print(x, y)
...
1 1
2 4
3 9

任何可迭代对象都可以作用于for循环,包括我们自定义的数据类型,只要符合迭代条件,就可以使用for循环。

2、迭代器

迭代是Python最强大的功能之一,是访问集合元素的一种方式,只能往前不会后退。
迭代器是一个可以记住遍历的位置的对象,字符串,列表或元组对象都可用于创建迭代器。迭代器有两个基本的方法:iter() 和 next()。

迭代器对象从集合的第一个元素开始访问,直到所有的元素被访问完结束。

1)iter()

字符串,列表或元组对象都可用于创建迭代器:

>>>list=[1,2,3,4] 
>>> it = iter(list) # 创建迭代器对象 
>>> print (next(it))# 输出迭代器的下一个元素 
1 
>>> print (next(it)) 
2 

迭代器对象可以使用常规for语句进行遍历:

#!/usr/bin/python3 
list=[1,2,3,4] 
it = iter(list) # 创建迭代器对象 
for x in it: 
    print (x, end=" ")

执行以上程序,输出结果如下:

1 2 3 4
02. next()

2)next() 函数

也可以使用 next() 函数:

#!/usr/bin/python3 
import sys 
# 引入 sys 模块
list=[1,2,3,4] 
it = iter(list) 
# 创建迭代器对象 
while True:
 try: 
print (next(it)) 
except 
StopIteration: sys.exit()
执行以上程序,输出结果如下:
1
2
3
4

3、生成器

在 Python 中,使用了 yield()函数被称为生成器(generator)。

跟普通函数不同的是,生成器是一个返回迭代器的函数,只能用于迭代操作,更简单点理解生成器就是一个迭代器。调用一个生成器函数,返回的是一个迭代器对象。

在调用生成器运行的过程中,每次遇到 yield 时函数会暂停并保存当前所有的运行信息,返回yield的值, 并在下一次执行next()方法时从当前位置继续运行。

调用一个生成器函数,返回的是一个迭代器对象。

以下实例使用 yield 实现斐波那契数列:

#!/usr/bin/python3 
import sys 
def fibonacci(n): 
# 生成器函数 - 斐波那契 
    a, b, counter = 0, 1, 0 
    while True: 
        if (counter > n): 
            return 
            yield a 
            a, b = b, a + b 
            counter += 1 

f = fibonacci(10) 
# f 是一个迭代器,由生成器返回生成 
while True: 
    try: 
        print (next(f), end=" ") 
    except 
        StopIteration: sys.exit()

执行以上程序,输出结果如下:

0 1 1 2 3 5 8 13 21 34 55

猜你喜欢

转载自blog.csdn.net/viatorsun/article/details/80213788
今日推荐