如何利用Python itertools.combinations逐个生成组合并保存至变量？

其实你用的itertools.combinations本身就是惰性迭代器——它不会一次性生成所有组合，而是在你需要的时候才生成下一个元素。你原来的for循环其实已经是在逐个获取组合了，只要稍作修改，就能把每个组合保存到变量里逐个处理。

方法1：用for循环逐个处理（最常用） #

直接遍历迭代器，每次把当前组合赋值给变量，再执行你的业务逻辑：

from itertools import combinations

l = [1, 2, 3, 4, 5]
# 生成组合迭代器，此时还没生成任何组合
comb_iterator = combinations(l, 2)

# 逐个取出组合并处理
for raw_pair in comb_iterator:
    current_pair = list(raw_pair)  # 保存到变量（转成列表方便操作）
    # 这里写你要对current_pair做的操作，比如计算、数据存储等
    print(f"正在处理组合: {current_pair}")

方法2：手动用next()获取下一个组合（适合手动控制流程） #

如果你需要更灵活地控制获取时机（比如按触发事件获取下一个），可以把迭代器存为变量，每次调用next()获取下一个组合：

from itertools import combinations

l = [1, 2, 3, 4, 5]
comb_iterator = combinations(l, 2)

# 手动获取第一个组合
first_pair = list(next(comb_iterator))
print(first_pair)  # 输出 [1, 2]

# 获取第二个组合
second_pair = list(next(comb_iterator))
print(second_pair)  # 输出 [1, 3]

# 遍历剩余所有组合（用异常处理结束迭代）
while True:
    try:
        pair = list(next(comb_iterator))
        print(f"当前组合: {pair}")
    except StopIteration:
        # 所有组合已取完，退出循环
        break

关键说明 #

迭代器的核心特性就是惰性生成，只有当你主动请求（比如for循环迭代、调用next()）时，才会生成下一个组合，不会一开始就把所有组合都加载到内存里，完全符合你“每次仅生成一个组合”的需求。

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