如何获取Python迭代器指向的当前元素且不移动迭代器？

Python原生迭代器的设计是单向前进的，仅支持next()方法推进并返回下一个元素，没有内置的current接口来获取当前指向的元素。要实现这个功能，核心思路是包装原迭代器，记录每次next()返回的元素，以下是两种可行方案：

方案一：自定义包装类 #

通过创建一个包装类，封装原迭代器并维护当前元素的状态：

class TrackedIterator:
    def __init__(self, iterator):
        self.iterator = iter(iterator)
        self.current_element = None
    
    def __next__(self):
        self.current_element = next(self.iterator)
        return self.current_element
    
    def __iter__(self):
        return self
    
    def current(self):
        if self.current_element is None:
            raise ValueError("迭代器尚未启动或已耗尽")
        return self.current_element

使用示例 #

# 用包装类处理原迭代对象
i = TrackedIterator([1,2,3,4])
first = next(i)  # 输出 1
second = next(i) # 输出 2
second_2 = i.current() # 输出 2
third = next(i) # 输出 3
third_2 = i.current() # 输出 3

方案二：闭包封装状态 #

用闭包封装迭代器和当前元素的状态，返回自定义的next和current函数：

def track_iterator(iterator):
    iterator = iter(iterator)
    current_val = None
    
    def next_val():
        nonlocal current_val
        current_val = next(iterator)
        return current_val
    
    def current_val_func():
        if current_val is None:
            raise ValueError("迭代器尚未启动或已耗尽")
        return current_val
    
    return next_val, current_val_func

使用示例 #

next_i, current_i = track_iterator([1,2,3,4])
first = next_i()  # 输出 1
second = next_i() # 输出 2
second_2 = current_i() # 输出 2
third = next_i() # 输出 3
third_2 = current_i() # 输出 3

注意事项 #

• 两种方案都需要处理迭代器未启动（未调用过next）或已耗尽（触发StopIteration后）的情况，避免返回无效值。
• 无法直接给Python内置迭代器（比如iter([1,2,3])生成的迭代器）动态添加current方法，必须通过包装的方式实现。

内容的提问来源于stack exchange，提问作者user171780