Python的循环与迭代器实现,快速查阅用。

 

本篇索引

(1)while循环与for循环

(2)一些迭代工具

(3)列表推导与序列解包

(4)迭代器

 

(1)while循环与for循环

while仅能用于普通循环,而for除了可以做循环外,还可以遍历序列、集合、字典、迭代器等。

需要注意的是,在类似:for i in somelist: 的循环中,迭代变量 i 的作用域并非for语句私有,循环结束后迭代变量依然保留最后一个值。有时候for可以用于直接迭代对象,如下2个例子所示:

 

直接迭代字典

for key in d:    # 直接对字典进行迭代相当于 for key in d.keys():
    print(key, d[key])

 

直接迭代文件

f = open('a.txt')
for line in f:    # 迭代文件对象中的每一行
    print(line)

 

● break

可用于跳出while或for循环。break和下面的continue语句仅应用于正在执行的最内层循环,如果要跳出多层嵌套循环结构,可使用raise()抛出异常。

 

● continue

结束本循环的当前轮,跳到本循环的下一轮开始。

 

● else

与while或for循环配对的else代码段,仅在本循环中没有调用过break时执行。

for i in range(0,100):
    if i > 100:
        break;
else:
    print('running else.')
    
# 结果为:running else.

 

 

 

(2)一些迭代工具

 

● 并行迭代

使用内建zip()函数,它可以把两个序列“压缩”在一起,返回一个元组列表。 当两个序列长度不等时,以短的那个迭代完就会停止。zip()还支持同时对多个(3个及以上)列表进行迭代。

a = 'abc'
b = [1,2,3]
for x,y in zip(a,b):
    print(x,y)
# 结果每次迭代依次显示:a 1、b 2、c 3

以上代码相当于:

for i in range(min(len(a),len(b)):
    print(a[i],b[i])

 

● 编号迭代

使用内建的enumerate()函数,可以在迭代的同时,同时获取迭代的次数索引。

a = 'abc'
for i,x in enumerate(a):
    print('i=%d,x=%s' %(i,x)) 
# 结果依次显示:i=0,x=a、i=1,x=b、i=2,x=c

 

 

 

(3)列表推导与序列解包

 

● 列表推导(list comprehension)

“列表推导”是利用for循环来快速创建新列表的一种方法。

 

a. 基本用法:

[ x运算表达式 for x in 迭代器 ]

举例:

[x*2 for x in range(5)]    # 结果生成新序列:[0,2,4,6,8]

 

b. 条件判断用法:

每次迭代时,只有当“条件表达式”的值为真时,才会对x进行运算,并将结果放入新序列。语法如下:

[ x运算表达式 for x in 迭代器 if 条件表达式 ]

举例:

[x for x in range(10) if x%2]    # 取出range(10)范围内的所有奇数,结果为:[1,3,5,7,9]

 

c. 完整用法:

可以同时使用多个嵌套的for循环并对每个循环使用if条件表达式,语法如下:

[ i_1,i_2,...,i_n运算表达式 for i_1 in iter1 if 条件表达式1 
                          for i_2 in iter2 if 条件表达式2
                          ...... 
                          for i_n in itern if 条件表达式n ]

以上语法大致相当于如下等价代码:

s = []
for i_1 in iter1:
    if 条件表达式1:
        for i_2 in iter2:
            if 条件表达式2:
                ......
                for i_n in itern:
                    if 条件表达式n:
                        s.append(i_1,i_2,...,i_n运算表达式)

举例:

a = 'abc'
b = [1,2,3]
[(i1,i2) for i1 in a 
         for i2 in b if i2 % 2]    
# 结果为:[('a',1), ('a',3), ('b',1), ('b',3), ('c',1), ('c',3) ]

 

 

 

● 迭代序列解包(sequence unpacking)

当可迭代对象返回元组时,这个特性比较有用。例如需要获取字典中的键-值对时,可使用序列解包方法:

for k,v in d.items():   # 每次循环同时取出字典中的键与值,分别放入k与v中
    print(k,v)

 

 

 

(4)迭代器

可以使一个普通对象也支持迭代操作,此时需要为这个对象实现__iter__()方法和__next__()方法。

 

● __iter__()方法:

本方法要求返回一个迭代器对象(通常可直接返回对象自身,详见下例)。

 

● __next__()方法:

__next__()方法由刚才通过__iter__()方法返回的迭代器对象进行调用,其作用是计算并返回迭代的下一个对象。当迭代器已迭代完所有的内容时,__next__()方法应该引发一个StopIteration异常,以通知外部的调用者结束迭代。

 

以下代码实现一个类似于 “列表[1,2,3,4,5]” 的迭代器:

class MyIterator:
    v = 0
    def __iter__(self):
        return self
    def __next__(self):
        self.v += 1
        if self.v > 5:
            raise StopIteration
        return self.v

myiter = MyIterator()
for i in myiter:
    print(i)
    
# 运行结果为,依次显示:1、2、3、4、5

 

● 迭代的外部原理:

上例中,当运行for语句进行迭代时,等效于运行以下代码:

_iter = myiter.__iter__()
while 1:
    try:
        i = _iter.__next__()
    except StopIteration:
        break
    print(i)  # 本句即为 for 循环中用户自己写的循环内容

从上面的代码中可以看到,迭代是使用__iter__()返回的对象来调用__next__()方法的,所以可以不一定是self。另外,内置函数iter()可以从可迭代的对象中获得迭代器,内置函数next()可以自动访问可迭代对象的__next__()方法。

因此,上面的第1行和第4行语句可等效为:

_iter = iter(s)    # 相当于 _iter = myiter.__iter__()
i = next(_iter)    # 相当于 i = _iter.__next__(),甚至直接写成:i = next(myiter) 也是可以的

 

● 将迭代和序列互相转换:

用list()的构造方法能显式地将迭代器转换为列,也能用内置函数iter()将序列转换为迭代器。

# ......
# 前面已定义MyIterator类

myiter = MyIterator()
s = list(myiter)
print(s)

# 结果为:[1,2,3,4,5]