Python遍历已耗尽生成器的行为、写法安全性及正确性咨询

问题背景 #

在Python 3环境下有一个可能产出单个或多个值的生成器，对应业务规则如下：

• 生成器产出的第一个值已经用于发起API请求，结果已完成存储，无需重复处理
• 从生成器的第二个值开始遍历，将遍历到的值传入API请求函数执行
• 如果生成器仅产出单个值，直接跳过循环逻辑；如果产出多个值，从第二个值开始执行循环内逻辑

针对单值生成器场景编写了如下测试代码，用print()模拟API请求逻辑：

def iterexample(): # 创建仅返回单个值的简单迭代器
    yield 0

print(0)
iter_obj = iterexample()
next(iter_obj) # 此处消费一次生成器，取走第一个值
for i in iter_obj: #1 生成器已耗尽
    print(i, ' inside loop')   #2 因为生成器已耗尽，该行被跳过
#3 循环外的其余代码会正常执行

测试运行结果符合预期：仅打印0，不会输出0 inside loop，运行输出为：

0

待解答疑问 #

1. 上述先调用next()取首个值、再遍历剩余生成器的写法是否安全、符合Pythonic规范？是否会抛出错误？
2. 该写法是否会产生无限循环？是否会出现API请求无限循环调用的问题？
3. 代码中标注的#1~#3三处注释对应的理解是否正确？

解答内容 #

• 关于写法安全性与规范性：
  这种写法是安全的，属于Python中跳过迭代器首元素的常规标准写法，完全符合Pythonic规范。在当前业务场景下（生成器至少产出1个值，第一个值已用于前置API请求），该写法不会抛出错误。
  唯一需要注意的边界场景：如果后续业务变更导致生成器可能为空（一个值都不产出），直接调用无默认值的next()会抛出StopIteration异常，这种情况可以给next()传入默认值做兼容，比如first_val = next(iter_obj, None)即可避免异常。
• 关于无限循环风险：
  该写法完全不会产生无限循环。生成器是一次性单向迭代的对象，手动调用next()取走第一个值后，后续for循环会严格按生成器的产出顺序遍历剩余值，所有值遍历完成后循环会自动终止，和普通遍历生成器的逻辑没有区别，不存在无限调用API的风险。
• 关于三处注释的正确性：
  对三处代码的理解全部正确：
  1. #1位置：单值生成器在被next()取走唯一的值后，确实已经处于耗尽状态，没有剩余可迭代内容
  2. #2位置：对已耗尽的生成器做for遍历时，循环体不会被执行，对应打印代码确实会被直接跳过
  3. #3位置：无论循环是否执行、执行多少次，循环体结束后，循环外的后续代码都会正常往下执行，不会被阻塞或跳过。

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