Python学习3---迭代器和可迭代对象

Python 迭代器（Iterators）是一种用于遍历可迭代对象的编程工具，例如列表、元组、字符串和字典等。迭代器允许你逐个访问可迭代对象的元素，而无需提前加载整个对象到内存中。本文将详细介绍 Python 中的迭代器、迭代对象、迭代协议和如何自定义迭代器。

一、可迭代对象（Iterable）
可迭代对象是一种包含多个元素的数据结构，你可以使用迭代器来遍历它们。常见的可迭代对象包括列表、元组、字符串、集合和字典等。可迭代对象可以通过内置的 iter() 函数来获取对应的迭代器。

my_list = [1, 2, 3, 4, 5]
my_iterable = iter(my_list)  # 获取列表的迭代器

二、迭代器（Iterator）
迭代器是一个带有 iter() 和 next() 方法的对象，用于在可迭代对象上进行迭代。__iter__返回迭代器本身，而__next__返回集合的下一个元素。当没有更多元素可供迭代时，迭代器会引发 StopIteration 异常。
下面是一个简单的迭代器示例：

class MyIterator:
    def __init__(self, start, end):
        self.current = start
        self.end = end

    def __iter__(self):
        return self

    def __next__(self):
        if self.current < self.end:
            result = self.current
            self.current += 1
            return result
        else:
            raise StopIteration

my_iterator = MyIterator(1, 6)
for num in my_iterator:
    print(num)

在这个示例中，MyIterator 类实现了迭代器协议，包括 iter() 和 next() 方法。迭代器会从 start 开始递增，直到达到 end 为止。

可迭代对象和迭代器的区别：
可迭代对象（如列表、字典）可以直接用于for循环，因为它们实现了__iter__方法，返回一个迭代器。
而迭代器是这些可迭代对象的实例，只能通过next()方法逐个访问元素。

三、内置迭代器
Python 提供了一些内置的迭代器，使遍历可迭代对象更加便捷。其中包括：
3.1、iter() 函数
iter() 函数用于获取可迭代对象的迭代器，然后用next()函数来获取下一个值。可以使用它来手动创建一个迭代器。

my_list = [1, 2, 3, 4, 5]
my_iterator = iter(my_list)

print(next(my_iterator))  # 输出: 1
print(next(my_iterator))  # 输出: 2
print(next(my_iterator))  # 输出: 3

3.2、enumerate() 函数
enumerate() 函数用于同时获取元素的索引和值，通常在循环中使用。

my_list = ["apple", "banana", "cherry"]
for index, value in enumerate(my_list):
    print(f"Index: {index}, Value: {value}")

3.3、zip() 函数
zip() 函数用于将多个可迭代对象按顺序配对在一起，创建一个包含元组的迭代器。

fruits = ["apple", "banana", "cherry"]
prices = [1.0, 0.5, 2.0]
for fruit, price in zip(fruits, prices):
    print(f"Fruit: {fruit}, Price: ${price}")

四、自定义迭代器
如果需要处理自定义数据结构或者更复杂的迭代逻辑，可以创建自己的迭代器类。一个迭代器类需要实现 iter() 和 next() 方法。

下面是一个示例，演示了如何创建一个自定义迭代器来生成斐波那契数列：

class FibonacciIterator:
    def __init__(self, limit):
        self.limit = limit
        self.previous = 0
        self.current = 1
        self.count = 0

    def __iter__(self):
        return self

    def __next__(self):
        if self.count < self.limit:
            result = self.current
            self.previous, self.current = self.current, self.previous + self.current
            self.count += 1
            return result
        else:
            raise StopIteration

fibonacci_sequence = FibonacciIterator(10)
for num in fibonacci_sequence:
    print(num)

在这个示例中，FibonacciIterator 类生成一个斐波那契数列，并限制了生成的元素个数。

五、迭代器的优点和适用场景
迭代器的优点在于它们可以处理大量数据，而不需要一次性加载整个数据结构到内存中，这有助于节省内存。迭代器还具有惰性求值的特性，只有在需要时才会计算下一个元素，从而提高了性能和效率。

5.1. 遍历大型文件或数据集
在处理大型文件或数据集时，将整个内容加载到内存中可能会导致内存不足的问题。迭代器的优点在于它们可以逐行或逐块地处理数据，而不需要一次性加载整个文件或数据集。

示例代码：

# 逐行读取大型文本文件并计算行数
line_count = 0
with open('large_file.txt', 'r') as file:
    for line in file:
        line_count += 1
print(f"Total lines in the file: {line_count}")

5.2. 处理无限序列
有些情况下，需要处理无限序列，例如斐波那契数列。使用迭代器可以轻松生成和处理这些无限序列。

示例代码：

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

    def __iter__(self):
        return self

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

# 创建一个生成斐波那契数列的迭代器
fibonacci_sequence = FibonacciIterator()
for i in range(10):
    print(next(fibonacci_sequence))

5.3. 生成无限流式数据
迭代器还可以用于生成无限流式数据，例如无限的自然数序列。

示例代码：

class NaturalNumbersIterator:
    def __init__(self):
        self.current = 1

    def __iter__(self):
        return self

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

# 创建一个生成自然数序列的迭代器
natural_numbers = NaturalNumbersIterator()
for i in range(10):
    print(next(natural_numbers))

5.4. 自定义数据结构的迭代行为
如果定义了自己的数据结构，可以通过实现 iter() 和 next() 方法来自定义其迭代行为。这样可以更好地控制如何遍历和访问对象的元素。

示例代码：

class MyCustomCollection:
    def __init__(self):
        self.data = [1, 2, 3, 4, 5]

    def __iter__(self):
        self.index = 0
        return self

    def __next__(self):
        if self.index < len(self.data):
            result = self.data[self.index]
            self.index += 1
            return result
        else:
            raise StopIteration

# 创建一个自定义数据结构的迭代器
my_collection = MyCustomCollection()
for item in my_collection:
    print(item)