如何将两个Python列表的全组合改为生成器以节省内存？

你提到的itertools库其实有专门处理这类笛卡尔积场景的函数——itertools.product，它本身是惰性生成器，不会一次性把所有组合加载到内存里，完全适配你的需求。另外也可以直接修改原代码的列表推导式，改成生成器表达式。

方法一：使用itertools.product（推荐） #

itertools.product会逐个生成两个列表的笛卡尔积元素，内存占用极低：

import itertools

list_a = [1,2,3,4,5,6,7,8,9,10]
list_b = [2,3,4,5,6,7,8,9,10,11]
all_combinations = itertools.product(list_a, list_b)  # 惰性生成器，仅按需生成元素
results = [Func(item) for item in all_combinations]

方法二：改用生成器表达式 #

如果不想引入额外库，直接把原列表推导式的方括号[]换成圆括号()，就变成了惰性求值的生成器表达式：

list_a = [1,2,3,4,5,6,7,8,9,10]
list_b = [2,3,4,5,6,7,8,9,10,11]
all_combinations = ((x,y) for x in list_a for y in list_b)  # 生成器表达式，不预存所有组合
results = [Func(item) for item in all_combinations]

额外优化：合并生成与计算逻辑 #

甚至可以跳过中间变量，直接把组合生成和函数调用合并，进一步简化代码：

# 直接生成结果列表，中间不存储任何组合
results = [Func((x,y)) for x in list_a for y in list_b]
# 如果结果也不需要一次性全部存储，可改成生成器
results_gen = (Func((x,y)) for x in list_a for y in list_b)

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