在上一篇文章
https://blog.csdn.net/nvd11/article/details/138738472
已经简单介绍了生成器 是 一种特殊的迭代器
而的确, 大部分普通的迭代器是可以被生成器取代的, 以达到简化代码的目的。
使用迭代器的例子
我们找回之前介绍迭代器用到的link list 例子:
https://blog.csdn.net/nvd11/article/details/138736024
里面的link list 保护了3个类
Node 类:
python
class Node:
def __init__(self, value):
self._value = value
self._next = None
@property
def value(self):
return self._value
@property
def next(self):
return self._next
@next.setter
def next(self, next):
self._next = next
@value.setter
def value(self, value):
self._value = value
LinkList类 它是iterable 可迭代对象
py
from loguru import logger
from src.iterator.sample_link_list.link_list_iterator import LinkListIterator
from src.iterator.sample_link_list.node import Node
class LinkList:
def __init__(self, first) -> None:
node = Node(first)
_first_node = node
_last_node = node
def __init__(self, *values) -> None:
if len(values) < 1:
raise ValueError("At least one node is required")
self._first_node = Node(values[0])
current = self._first_node
for i in range(1, len(values)):
current.next = Node(values[i])
current = current.next
self._last_node = current
def __iter__(self):
return LinkListIterator(self._first_node)
# to print all nodes's value but not nodes themselves
def print_nodes(self):
current = self._first_node
while current:
logger.info(current.value)
current = current.next
def get_length(self):
current = self._first_node
count = 0
while current:
count += 1
current = current.next
return count
def append(self, value):
if self.get_length() == 0:
self._first_node = Node(value)
self._last_node = self._first_node
else:
self._last_node.next = Node(value)
self._last_node = self._last_node.next
可以见到, 它的__iter__ 方法返回1个迭代器对象
LinkListIterator(self._first_node)
所以我们还需要被编写这个迭代器的类代码:
它要实现__next__ 方法
python
class LinkListIterator:
def __init__(self, _first_node) -> None:
self._current_node = _first_node
def __iter__(self):
return self
def __next__(self):
if not self._current_node:
raise StopIteration
current = self._current_node
self._current_node = current.next
return current.value
改成用生成器
一旦我们改用生成器, 则只需要让LinkList 这个interable 的__iter__ 方法变成1个生成器即可
如何让1个方法/函数变成1个生成器, 上一篇文章讲过, 就是使用yield 关键字啊
修改后的代码如下:
python
def __iter__(self):
current = self._first_node
while current:
yield current.value
current = current.next
return
注意这个生成器方法默认是没有参数的
所以它第一元素是 self_first_node
之后yield 当前node 的value出去 , 然后不断让当前元素变成下个原属就好
而这时 原来了的LinkListIterator 就不再需要,大大简化了代码!