如何解决itertools.product()引用传递导致重复插入0的问题？

你遇到的问题核心是：itertools.product返回的是原可迭代对象中元素的引用，而非独立副本。比如你传入的[[1],[2]]，每次循环拿到的p[0]其实就是原始的[1]或[2]列表本身。第一次循环修改[1]为[0,1]，第二次循环再拿到这个列表时，它已经被修改过，再执行insert(0,0)就会变成[0,0,1]，这就是重复插入0的原因。

下面是两种可行的解决方案：

方案1：生成product时提前创建元素副本 #

在传入product之前，先给每个列表元素创建独立副本，这样后续修改不会影响原始列表和其他迭代项：

from itertools import product

# 对第一个可迭代对象的每个元素创建副本
for p in product([x.copy() for x in [[1], [2]]], [[4], [5]]):
    p[0].insert(0, 0)
    print(p)

预期输出：

([0, 1], [4])
([0, 1], [5])
([0, 2], [4])
([0, 2], [5])

方案2：循环内操作前创建副本 #

如果不想改动传入product的参数，可以在每次循环中先复制要修改的列表，再执行插入操作：

from itertools import product

for p in product([[1], [2]], [[4], [5]]):
    # 复制p[0]得到独立新列表，也可以用p[0][:]实现
    modified_list = p[0].copy()
    modified_list.insert(0, 0)
    # 用新列表和原p[1]组成元组输出
    print((modified_list, p[1]))

预期输出和方案1一致。

注意：如果处理的是嵌套多层的列表，需要使用copy.deepcopy()来创建深副本，避免嵌套元素仍为引用的问题。你的场景是一维列表，用copy()或切片就足够。

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