如何在while循环中仅保留含新增元素的it.combinations()结果？

你的问题戳中了这类循环的核心痛点——每次新增节点后完全没必要重新遍历所有历史组合，毕竟那些不含新节点的组合之前已经处理过了。我们只需要精准生成包含至少一个新增节点的8元组合，就能大幅减少循环次数，直接提升运行效率。

具体思路拆解 #

每次循环你会新增两个节点，我们可以把这两个节点单独提取出来，然后分三种互斥的情况生成需要处理的组合：

1. 仅包含第一个新节点的组合：从新增前的旧元素集合里选7个元素，和第一个新节点凑成8元组
2. 仅包含第二个新节点的组合：同理，从旧元素集合里选7个，和第二个新节点凑成8元组
3. 同时包含两个新节点的组合：从旧元素集合里选6个，加上这两个新节点组成8元组

这三种情况完全覆盖了所有需要新处理的组合，不会漏掉也不会重复，完美跳过之前已经检查过的旧组合。

修改后的代码示例 #

import itertools as it

h = 0
while len(added) < i + 1:
    # 先记录当前good的长度（新增前的元素数量）
    old_length = len(good)
    # 生成并添加新节点
    new_node1 = z1 + h*2
    new_node2 = z2 + h*2
    g.add_node(new_node1)
    g.add_node(new_node2)
    good.append(new_node1)
    good.append(new_node2)
    
    # 获取新增前的旧元素集合（无需复制，直接切片更高效）
    old_good = good[:old_length]
    
    # 1. 处理仅包含new_node1的组合
    for u_rest in it.combinations(old_good, 7):
        u1 = list(u_rest) + [new_node1]
        # 执行你的剩余代码...
    
    # 2. 处理仅包含new_node2的组合
    for u_rest in it.combinations(old_good, 7):
        u1 = list(u_rest) + [new_node2]
        # 执行你的剩余代码...
    
    # 3. 处理同时包含两个新节点的组合
    for u_rest in it.combinations(old_good, 6):
        u1 = list(u_rest) + [new_node1, new_node2]
        # 执行你的剩余代码...
    
    h += 1

额外优化说明 #

• 用good[:old_length]获取旧元素集合比copy()更高效，因为切片操作在Python里是轻量级的，不需要额外复制内存。
• 三种组合情况是互斥的，不会出现重复处理同一个组合的问题，逻辑上完全等价于“所有包含新增节点的8元组合”。
• 如果你的原有代码里对组合有过滤条件，直接套在这三个循环里即可，业务逻辑不需要做任何改动。

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