Python二维列表筛选元素遗漏问题：如何修复i+j+k≠n的逻辑

问题根源 #

你在遍历列表的同时直接删除元素，会导致列表长度动态缩短，后续元素的索引向前偏移，循环会跳过部分需要删除的元素。比如原列表中[1,1,0]这类符合删除条件的元素，会因为前面元素被删除后索引错位而被遗漏。

原问题代码示例 #

假设你的代码大致如下：

x, y, z, n = 2, 2, 2, 2
# 生成所有i、j、k的组合
all_combinations = []
for i in range(x + 1):
    for j in range(y + 1):
        for k in range(z + 1):
            all_combinations.append([i, j, k])
# 遍历删除i+j+k=n的元素
for item in all_combinations:
    if sum(item) == n:
        all_combinations.remove(item)
print(all_combinations)

运行后会发现部分符合条件的元素（如[1,1,0]）未被移除。

修复方案 #

方案1：列表推导式直接生成目标列表（推荐）

直接在生成组合阶段过滤掉不符合条件的元素，无需后续删除操作，效率更高：

x, y, z, n = 2, 2, 2, 2
result = [
    [i, j, k]
    for i in range(x + 1)
    for j in range(y + 1)
    for k in range(z + 1)
    if i + j + k != n
]
print(result)

方案2：遍历列表的副本

遍历原列表的切片副本，这样删除原列表元素时不会影响遍历顺序：

x, y, z, n = 2, 2, 2, 2
all_combinations = []
for i in range(x + 1):
    for j in range(y + 1):
        for k in range(z + 1):
            all_combinations.append([i, j, k])
# 遍历副本而非原列表
for item in all_combinations[:]:
    if sum(item) == n:
        all_combinations.remove(item)
print(all_combinations)

方案3：倒序遍历原列表

倒序遍历索引，删除前面的元素不会影响后续待处理元素的索引：

x, y, z, n = 2, 2, 2, 2
all_combinations = []
for i in range(x + 1):
    for j in range(y + 1):
        for k in range(z + 1):
            all_combinations.append([i, j, k])
# 从最后一个元素倒序遍历
for idx in range(len(all_combinations)-1, -1, -1):
    if sum(all_combinations[idx]) == n:
        del all_combinations[idx]
print(all_combinations)

效果验证 #

三种方案都能正确移除所有满足i+j+k=n的元素，返回结果与预期一致。

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