Python中return和yield的区别

在 Python 中，return 和 yield 都用于从函数中返回值，但它们的机制和用途完全不同：

1. return

• 功能：return 用于从函数中返回一个值，并立即终止函数的执行。
• 执行方式：调用函数时，遇到 return 会直接返回结果，函数状态被销毁，下次调用重新开始。
• 适用场景：普通函数（一次性计算并返回结果）。
• 示例：

  def add(a, b):
      return a + b
  
  result = add(2, 3)  # 返回 5，函数执行结束
  

2. yield

• 功能：yield 用于定义生成器函数，会暂停函数执行并保留当前状态，下次调用时从暂停处继续。
• 执行方式：函数返回一个生成器对象，每次调用 next() 或迭代时执行到 yield 返回一个值，函数状态被保留。
• 适用场景：需要惰性计算（如处理大数据流或无限序列）。
• 示例：

  def count_up_to(n):
      i = 0
      while i < n:
          yield i  # 每次 yield 返回 i，函数暂停
          i += 1
  
  gen = count_up_to(3)
  print(next(gen))  # 输出 0
  print(next(gen))  # 输出 1（从上次暂停处继续）
  

关键区别

特性	return	yield
返回值后状态	函数终止，状态销毁	函数暂停，状态保留
返回类型	直接返回结果	返回生成器对象（可迭代）
内存效率	一次性返回所有数据	惰性计算，节省内存
典型用途	普通函数	生成器、大数据处理、协程

示例对比

# 使用 return（一次性计算）
def squares_return(n):
    result = []
    for i in range(n):
        result.append(i * i)
    return result  # 返回完整列表（占用内存）

# 使用 yield（按需生成）
def squares_yield(n):
    for i in range(n):
        yield i * i  # 每次生成一个值

# 调用对比
print(squares_return(5))  # 输出 [0, 1, 4, 9, 16]
gen = squares_yield(5)
print(next(gen))  # 输出 0（惰性计算）

总结

• 用 return 当需要立即返回所有结果。
• 用 yield 当需要逐步生成结果（如处理大文件、流数据或无限序列）。