Python 中如何使用 enumerate 函数创建自定义迭代器

2023-03-31 00:00:00 函数 自定义 如何使用

在Python中，可以使用enumerate函数将一个可迭代对象转换为索引-元素对。如果想要创建自定义迭代器，则需要定义一个迭代器类，并在该类中实现__iter__和__next__方法。

下面是一个演示如何使用enumerate函数创建自定义迭代器的例子：

class CustomIterator:
    def __init__(self, iterable):
        self._iterable = iterable
        self._index = 0

    def __iter__(self):
        return self

    def __next__(self):
        if self._index >= len(self._iterable):
            raise StopIteration
        value = self._iterable[self._index]
        self._index += 1
        return self._index, value

words = ["pidancode.com", "is", "a", "website", "for", "learning", "Python", "programming"]
custom_iterator = CustomIterator(words)

for index, word in custom_iterator:
    print(f"{index}: {word}")

在上面的代码中，我们定义了一个名为CustomIterator的迭代器类。该类接受一个可迭代对象作为参数，并将该可迭代对象转换为索引-元素对的形式。

在__init__方法中，我们将传入的可迭代对象保存在_iterable属性中，并初始化一个索引变量_index，用于跟踪当前元素的位置。

在__iter__方法中，我们返回实例自身，因为该实例本身就是迭代器对象。

在__next__方法中，我们检查当前索引是否超出了可迭代对象的长度。如果当前索引已经超出了可迭代对象的长度，则抛出StopIteration异常。否则，我们从可迭代对象中获取当前索引处的元素，并将_index变量加1，以便在下一次迭代时获取下一个元素。

在上面的代码中，我们使用自定义迭代器来遍历words列表，并在每次迭代时打印出索引和元素的值。运行上述代码，可以得到以下
输出结果：

1: pidancode.com
2: is
3: a
4: website
5: for
6: learning
7: Python
8: programming

需要注意的是，上述代码中的迭代器类仅是一个简单的示例，可以根据具体需求来修改该类的实现。

相关文章