如何正确使用生成器表达式实现多层数据结构的扁平化处理-Python教程-PHP中文网

本文详解生成器表达式在嵌套迭代场景中的常见误区，重点说明为何 (word for word in split_lines) 无法扁平化二维结构，并提供标准的嵌套生成器写法及优化方案。

本文详解生成器表达式在嵌套迭代场景中的常见误区，重点说明为何 `(word for word in split_lines)` 无法扁平化二维结构，并提供标准的嵌套生成器写法及优化方案。

生成器表达式是 Python 中高效处理流式数据的重要工具，但其行为高度依赖对“迭代对象层级”的准确理解。在文件逐行读取并拆分为单词的典型任务中，一个常见错误是误将二维结构（即“行列表 → 每行的词列表”）当作一维序列直接展开。

例如，以下代码看似合理，实则逻辑有误：

lines = open("file.txt")
split_lines = (line.split() for line in lines)  # 生成器：产出 ['word1', 'word2'], ['word3'], ...
words = (word for word in split_lines)          # ❌ 错误！此处 word 是整个列表，而非单个字符串
for word in words:
    print(word)  # 输出：['hello', 'world']、['python', 'is', 'great'] —— 并非单个单词

问题根源在于：split_lines 是一个产出 list 对象的生成器（每个 list 是一行分割后的单词数组），而 (word for word in split_lines) 仅执行一层迭代，等价于 for word in [list1, list2, ...]，因此 word 实际上是列表，而非字符串。

✅ 正确做法是使用嵌套生成器表达式（即双重 for），显式展开内层结构：

lines = open("file.txt")
split_lines = (line.split() for line in lines)
words = (word for line in split_lines for word in line)  # ✅ 正确：先遍历 split_lines，再遍历每个 line
for word in words:
    print(word)  # 输出：'hello'、'world'、'python'、'is'、'great'...

该语法等价于以下嵌套循环：

for line in split_lines:
    for word in line:
        yield word

更进一步，可将逻辑合并为单层生成器，提升简洁性与内存效率：

with open("file.txt") as lines:  # ✅ 推荐：使用 with 确保文件自动关闭
    words = (word for line in lines for word in line.split())
    for word in words:
        print(word)

⚠️ 注意事项：