【送面试题】探索Python中的迭代器（Iterator）和可迭代对象（Iterable）

class MyIterable:
    def __init__(self, data):
        self.data = data

    def __iter__(self):
        return iter(self.data)

# 创建一个可迭代对象
my_iterable = MyIterable([1, 2, 3, 4, 5])

# 使用for循环遍历可迭代对象
for item in my_iterable:
    print(item)

在上面的代码中，我们定义了一个名为MyIterable的类，并实现了__iter__()方法。该方法返回一个迭代器对象，它可以用于遍历MyIterable实例中的数据。

迭代器（Iterator）

迭代器是一种实现了迭代器协议（Iterator Protocol）的对象。迭代器协议要求迭代器对象实现__iter__()和__next__()两个方法。

__iter__()方法返回迭代器对象自身。
__next__()方法返回迭代器中的下一个元素。如果没有下一个元素，它会引发StopIteration异常。

下面是一个示例代码，展示了如何创建一个简单的迭代器：

class MyIterator:
    def __init__(self, data):
        self.data = data
        self.index = 0

    def __iter__(self):
        return self

    def __next__(self):
        if self.index >= len(self.data):
            raise StopIteration
        item = self.data[self.index]
        self.index += 1
        return item

# 创建一个迭代器
my_iterator = MyIterator([1, 2, 3, 4, 5])

# 使用for循环遍历迭代器
for item in my_iterator:
    print(item)

在上面的代码中，我们定义了一个名为MyIterator的类，并实现了__iter__()和__next__()方法。在__next__()方法中，我们通过self.index来追踪当前遍历的位置，并返回相应的元素。当没有下一个元素时，我们引发StopIteration异常来结束迭代。

迭代器和可迭代对象的关系

在前面的示例代码中，我们可以观察到迭代器和可迭代对象之间的关系。事实上，可迭代对象和迭代器之间存在紧密的联系。

可迭代对象是一种具有__iter__()方法的对象，它返回一个迭代器对象。而迭代器对象本身也是可迭代的，它实现了__iter__()方法，返回自身，并且还实现了__next__()方法（续上文）

来返回下一个元素。因此，迭代器既是可迭代对象，也是迭代器本身。

扫描二维码关注公众号，回复： 17364557 查看本文章

迭代器和可迭代对象的这种关系使得我们可以使用统一的方式来处理不同类型的数据集合。无论是列表、元组、集合还是自定义的数据结构，只要它们实现了迭代器协议，我们就可以使用相同的语法来进行遍历。

应用示例

迭代器和可迭代对象的概念在实际编程中非常实用。下面是几个示例，展示了它们在不同场景下的应用。

示例一：自定义可迭代对象

假设我们要处理一个非常大的数据集合，但是由于内存限制，我们无法一次性将所有数据加载到内存中。这时，我们可以使用自定义的可迭代对象来逐个读取数据，从而避免内存溢出的问题。

class LargeDataset:
    def __init__(self, filepath):
        self.filepath = filepath

    def __iter__(self):
        with open(self.filepath, 'r') as file:
            for line in file:
                yield line.strip()

# 创建一个可迭代对象
dataset = LargeDataset('large_dataset.txt')

# 使用for循环逐行处理数据
for item in dataset:
    process_data(item)

在上面的代码中，我们定义了一个名为LargeDataset的类，它代表了一个大型的数据集合。在__iter__()方法中，我们使用yield关键字来生成每一行的数据，并逐行返回。这样，我们就可以通过迭代的方式逐个读取数据行，而不需要将整个数据集合一次性加载到内存中。

示例二：实现斐波那契数列的迭代器

斐波那契数列是一个经典的数学问题，可以使用迭代器来生成斐波那契数列的下一个元素。

class FibonacciIterator:
    def __init__(self):
        self.current = 0
        self.next = 1

    def __iter__(self):
        return self

    def __next__(self):
        result = self.current
        self.current, self.next = self.next, self.current + self.next
        return result

# 创建一个斐波那契数列的迭代器
fibonacci = FibonacciIterator()

# 打印前十个斐波那契数
for _ in range(10):
    print(next(fibonacci))

在上面的代码中，我们定义了一个名为FibonacciIterator的迭代器类。在__next__()方法中，我们使用两个变量current和next来记录当前和下一个斐波那契数。每次调用__next__()方法时，我们返回当前的斐波那契数，并更新current和next的值。

通过使用斐波那契数列的迭代器，我们可以方便地生成任意长度的斐波那契数列。

结论

迭代器和可迭代对象是Python语言中的重要概念，它们为我们提供了一种简洁而有效的方式来处理数据集合。可迭代对象允许我们使用for循环来遍历数据集合，而迭代器则进一步扩展了遍历的方式，允许我们按需逐个返回数据元素。

在本文中，我们深入探讨了迭代器和可迭代对象的概念，介绍了它们的工作原理，并通过示例代码展示了它们在实际编程中的应用。通过理解迭代器和可迭代对象的原理和用法，我们可以更加灵活地处理数据集合，并编写出更加高效和可维护的Python