Python 的内置函数 `next`

next() 是 Python 的一个内置函数，用于从迭代器中获取下一个元素。它是处理迭代器和生成器时非常重要的工具函数。

基本用法

next(iterator[, default])

iterator: 要从中获取下一个元素的迭代器对象
default (可选): 当迭代器耗尽时返回的默认值

详细说明

1. 基础示例

numbers = iter([1, 2, 3])
print(next(numbers))  # 输出: 1
print(next(numbers))  # 输出: 2
print(next(numbers))  # 输出: 3

2. 处理迭代器耗尽

当迭代器耗尽时，next() 会抛出 StopIteration 异常：

numbers = iter([1])
print(next(numbers))  # 输出: 1
print(next(numbers))  # 抛出 StopIteration 异常

可以指定默认值来避免异常：

numbers = iter([1])
print(next(numbers, 'No more items'))  # 输出: 1
print(next(numbers, 'No more items'))  # 输出: No more items

3. 与生成器一起使用

next() 经常用于控制生成器的执行：

def count_up_to(n):
    i = 1
    while i <= n:
        yield i
        i += 1

counter = count_up_to(3)
print(next(counter))  # 输出: 1
print(next(counter))  # 输出: 2
print(next(counter))  # 输出: 3

4. 实际应用场景

文件读取：逐行读取大文件时避免一次性加载全部内容
数据处理：处理大型数据集时按需获取数据
无限序列：处理斐波那契数列等无限序列
协程控制：在异步编程中控制协程执行

5. 实现原理

next() 函数实际上调用了迭代器的 __next__() 方法。对于自定义迭代器类，需要实现 __next__() 方法才能与 next() 函数配合使用。

class CountDown:
    def __init__(self, start):
        self.current = start
    
    def __iter__(self):
        return self
    
    def __next__(self):
        if self.current <= 0:
            raise StopIteration
        else:
            self.current -= 1
            return self.current + 1

counter = CountDown(3)
print(next(counter))  # 输出: 3
print(next(counter))  # 输出: 2
print(next(counter))  # 输出: 1

注意事项

确保传递给 next() 的对象确实是迭代器（实现了 __next__() 方法）
处理可能出现的 StopIteration 异常，或者使用默认值参数
对于不可迭代对象（如整数、字符串本身），需要先使用 iter() 转换为迭代器
多次调用 next() 会消耗迭代器中的元素，不可逆

Python 的内置函数 next