为什么部分Python迭代器耗尽后仍可执行求和操作？

你遇到的差异和sum函数的实现无关，核心问题是你自定义的迭代器违反了Python迭代器协议的约定。

基础逻辑说明 #

Python中所有接收可迭代对象的内置函数（包括sum、max、min等），执行时都会先调用iter(传入对象)获取迭代器，再不断调用next()取值直到遇到StopIteration异常终止。

内置列表迭代器i1的行为逻辑 #

原生列表返回的迭代器严格遵守迭代器协议：

• 它的__iter__方法仅会返回自身，不会修改内部的游标状态
• 第一次调用sum(i1)已经将迭代器的游标移动到末尾，后续不管是直接调用next(i1)还是再次调用sum(i1)触发iter(i1)，拿到的都是已经耗尽的迭代器，所以求和结果为0，调用next()也会抛出异常。

自定义迭代器i2的异常原因 #

你写的CustomIterator的__iter__方法违反了协议约定：每次调用__iter__时都会主动重置内部计数器self.n = 0。

• 每次调用sum(i2)时，第一步的iter(i2)就会触发__iter__执行，把计数器重置为0，接下来遍历自然能拿到1、2、3三个值，求和得到6
• 你单独调用next(i2)时不会触发__iter__方法，直接调用__next__时计数器还是之前的4，加1后大于3就抛出StopIteration，和你观察到的现象完全吻合

修复方案 #

将计数器初始化放到构造方法中，__iter__仅返回自身，修改后自定义迭代器的行为就会和内置迭代器完全一致：

class CustomIterator:
  def __init__(self):
    # 计数器仅在实例化时初始化一次
    self.n = 0
  
  def __iter__(self):
    # __iter__仅返回自身，不修改状态
    return self

  def __next__(self):
    self.n += 1
    if self.n > 3:
      raise StopIteration
    return self.n

补充说明 #

你当前实现的CustomIterator其实更接近「可迭代对象」而非「标准迭代器」：普通可迭代对象（比如原生list）每次调用iter()都会返回一个全新的迭代器，你现在的实现相当于每次调用__iter__都把当前实例重置为新迭代器的状态，才导致了不符合常规迭代器的表现。

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