python的迭代器---Iterable和iterator
迭代器—Iterable和iterator
举例
test_list = [1, 2, 3]
test_dump = {"name": "zhangsan", "age": "12"}
for i in test_list:
print(i)
print(type(i))
for key in test_dump:
print(key)
print(test_dump[key])
print(type(key))
结果
1
2
3
name
zhangsan
age
12
Iterable和iterator
先说概念
Iterable 可迭代对象，for loop的 in 后面必须是一个Iterable 数据的保存者，可以没有状态，他不必知道iterator数到了哪里，但是必须有能力产生一个iterator
一个Iterable，要么有__iter__这个method，要么就需要是一个sequence（顺序排列，队列），有__getitem__这个method，这两个都是为了保证可以返 回一个iterator。
iterator 迭代器，表示数据流的对象，iterator必须是有状态的，但是并不需要实现一个Iterable，一个iterator必须要有__next__这个方法，这个method保证他在被 next作用的时候，可以返回下一个Iterable里的值，也就是一个新的iterator
for loop的时候会先对Iterable进行一个iter()操作，然后再根据iterator进行操作，加下来我们来理解一下上面这些概念的意思
class NodeIter:
def __init__(self, node):
self.curr_node = node
def __next__(self):
if self.curr_node is None:
# 如果数据已经到头了，就需要抛出错误
raise StopIteration
# 没有到头的话，就返回这个node，并且把下一个值存储
node, self.curr_node = self.curr_node, self.curr_node.next
return node
class Node:
def __init__(self, name):
self.name = name
self.next = None
def __iter__(self):
# Node是一个Iterable，但是他的__iter__方法，返回了一个iterator
return NodeIter(self)
node1 = Node("Node1")
node2 = Node("Node2")
node3 = Node("Node3")
node1.next = node2
node2.next = node3
for node in node1:
# 等于for node in iter(node1)，向Iterable要了一个iterator
print(node.name)
结果
Node1
Node2
Node3
但是呢官方推荐这个iterator除了有__next__函数之外，还需要有__iter__函数，是为了让每一个iterator也是一个Iterable，就是让每一个迭代器也是可迭代的
class NodeIter:
def __init__(self, node):
self.curr_node = node
def __next__(self):
if self.curr_node is None:
# 如果数据已经到头了，就需要抛出错误
raise StopIteration
# 没有到头的话，就返回这个node，并且把下一个值存储
node, self.curr_node = self.curr_node, self.curr_node.next
return node
class Node:
def __init__(self, name):
self.name = name
self.next = None
def __iter__(self):
# Node是一个Iterable，但是他的__iter__方法，返回了一个iterator
return NodeIter(self)
node1 = Node("Node1")
node2 = Node("Node2")
node3 = Node("Node3")
node1.next = node2
node2.next = node3
# 先求了一个iterator出来
it = iter(node1)
# 将iterator的第一个值取出来了
first = next(it)
# 再来迭代it这个iterator
for node in it:
# 等于for node in iter(node1)
print(node.name)
报错
TypeError: 'NodeIter' object is not iterable
看起来是不是很难受
class NodeIter:
def __init__(self, node):
self.curr_node = node
def __next__(self):
if self.curr_node is None:
# 如果数据已经到头了，就需要抛出错误
raise StopIteration
# 没有到头的话，就返回这个node，并且把下一个值存储
node, self.curr_node = self.curr_node, self.curr_node.next
return node
# 解决办法很简单，就是在iterator里也实现一个__iter__方法，返回他自己就可以了
def __iter__(self):
return self
class Node:
def __init__(self, name):
self.name = name
self.next = None
def __iter__(self):
# Node是一个Iterable，但是他的__iter__方法，返回了一个iterator
return NodeIter(self)
node1 = Node("Node1")
node2 = Node("Node2")
node3 = Node("Node3")
node1.next = node2
node2.next = node3
# 先求了一个iterator出来
it = iter(node1)
# 将iterator的第一个值取出来了
first = next(it)
# 再来迭代it这个iterator
for node in it:
# 等于for node in iter(node1)
print(node.name)
结果
Node2
Node3