for循环的本质
我们常常会写如下的代码
a = [1,2,3]
for item in a:
print(item)
for i in range(10):
print(i)
for i, item in enumerate(a):
print(i, item)
第一部分代码例子很好理解,循环遍历列表 a 里面的元素。 python 的 for … in … 语义非常清晰。
第二部分代码例子和第一部分比较起来,似乎 range(10) 和列表等价。
第三部分代码例子和第一部分比较起来, enumerate(a) 每次迭代会返回两个元素,第一个 i 代表序号。
究竟是不是这样呢。 python 这几个关键字是怎么让其在迭代用法这么灵活呢(想想java 等其他语言吧)。
有部分读者可能有了解过。迭代器协议
迭代器协议:
- 实现
__iter__
方法, 该方法返回迭代器对象。 - 实现
__next__
方法,该方法返回迭代的值,抛出 StopIteration 表示停止迭代。
可能还是云里雾里的,没关系,我们直接看一个例子。
class A:
def __init__(self, end):
self.i = -1
self.end = end
def __iter__(self):
return self
def __next__(self):
self.i += 1
if self.i < self.end:
return self.i
else: