PyYAML加载generator object迭代后内容消失的原因咨询

嗨，这个问题其实和PyYAML的关系不大，核心是生成器（generator）对象本身的特性导致的，我来给你讲明白：

1. 生成器的“一次性”本质 #

生成器是Python里一种惰性求值的对象，它不会像列表那样把所有数据提前加载到内存里，而是在迭代过程中逐个生成元素。打个比方，它就像一个只能往外倒一次的零食袋，倒完最后一颗就空了——当你第一次用for循环或者next()函数迭代它时，它会一步步生成所有元素，等所有元素都被取完之后，再去迭代它就不会返回任何内容了，哪怕变量本身还是一个生成器对象，里面已经没有可生成的数据了。

2. PyYAML为什么返回生成器？ #

你应该是用了yaml.load_all()或者新版的yaml.safe_load_all()来加载你的template.yml文件吧？这个方法是专门用来处理包含多个YAML文档的文件（文档之间用---分隔），它返回生成器的目的是为了高效处理大文件——不用一次性把所有文档都读进内存，而是需要的时候再生成下一个文档的内容。

举个你可能遇到的代码场景：

import yaml

with open('template.yml', 'r') as f:
    content = yaml.safe_load_all(f)

# 第一次迭代，正常输出所有文档内容
for doc in content:
    print(doc)

# 第二次迭代，完全没有输出
for doc in content:
    print(doc)  # 啥也不会打印

解决办法 #

根据你的需求，有两种常见的处理方式：

• 把生成器转成列表：如果需要反复使用这些数据，把生成器转换成列表，这样所有数据会被一次性加载到内存里，就可以多次迭代了：

  import yaml
  
  with open('template.yml', 'r') as f:
      content = list(yaml.safe_load_all(f))  # 转成列表存储
  
  # 现在可以随便迭代多少次
  for doc in content:
      print(doc)
  
  for doc in content:
      print(doc)  # 依然能正常输出
  

• 用单个文档的加载方法：如果你的template.yml里只有一个YAML文档，直接用yaml.safe_load()（不要用load_all()），它会直接返回单个Python对象（比如字典、列表），自然就不会有“一次性迭代”的问题了：

  import yaml
  
  with open('template.yml', 'r') as f:
      content = yaml.safe_load(f)  # 直接返回单个对象
  
  # 可以随意访问content里的内容
  print(content)
  

简单总结一下：不是变量被“清空”了，而是生成器已经被耗尽了，这是它的设计特性。转换成列表或者改用单个文档的加载方法，就能解决你的问题啦～

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