部分场景下yield执行速度慢于return吗？其性能优势有何前提？

首先明确结论：你的观察是正确的，yield的执行速度优势确实绝大多数情况下都出现在不需要遍历完全部迭代的场景，除此之外仅在超大结果集的场景下yield会因为内存占用优势获得更高的实际运行效率。

为什么字典清理场景下return更快 #

你做字典清理时需要拿到全量的清理结果，这时候两种实现的执行逻辑差异是：

• return版本：一次性在函数内部构造好完整的清理后字典，直接返回整个对象，仅产生一次函数调用开销
• yield版本：返回的是生成器对象，你需要额外遍历整个生成器才能把所有键值对组装成字典，每一次迭代都要付出生成器上下文切换的额外开销，最终耗时自然比return更高

yield的性能优势场景 #

yield更快的场景只有两类：

1. 不需要遍历完所有迭代的场景：比如筛选字典中第一个符合要求的键值对、只取前N个处理结果等，yield可以在拿到需要的结果后直接终止后续的所有迭代操作，不需要像return版本一样必须先处理完所有元素再返回结果，迭代总量越大、终止时机越早，yield的优势越明显。
2. 结果集远超内存容量的场景：如果要处理的原始字典有百万甚至千万级条目，return版本一次性构造全量结果会占用大量内存，甚至触发内存交换导致运行卡顿，yield每次仅返回单个键值对，内存占用始终维持在极低水平，这种场景下哪怕需要全量遍历，yield的实际运行速度也会高于return。

代码示例对比 #

# return版字典清理
def clean_dict_return(raw_dict):
    cleaned = {}
    for k, v in raw_dict.items():
        if v is not None:
            cleaned[k] = v
    return cleaned

# yield版字典清理
def clean_dict_yield(raw_dict):
    for k, v in raw_dict.items():
        if v is not None:
            yield k, v

• 全量取结果场景：dict(clean_dict_yield(raw)) 耗时高于 clean_dict_return(raw)，符合你的测试结果
• 仅取第一个符合要求的结果场景：

  # yield版本，找到第一个结果后直接终止，后续元素无需处理
  for k, v in clean_dict_yield(raw):
      target = v
      break
  # return版本，必须处理完所有元素才能拿到第一个结果
  target = next(iter(clean_dict_return(raw).values()))
  

  该场景下yield的耗时会远低于return版本。

你提供的性能对比图也验证了上述逻辑：

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