Python 数组意外覆盖

Question

我有一个关于 python 列表的问题，当我尝试将列表添加到列表时，我想生成一个类似 (001, 002, 011, 012) 的数组，新列表尝试覆盖旧元素

n = 3
def setNextCombination(P):
    idx = len(P) - 1
    while True:       
        if idx == 0:
            break
        if P[idx] < idx:
            P[idx] += 1
            curr = idx + 1
            while curr < len(P):
                P[curr] = P[idx]
                curr += 1
            return P, True
        idx -= 1
    return [], False

def path_choosing():
    P = np.zeros(n, dtype=int)
    flag = True
    arr = [0,0,0,0]
    idx = 0
    while True:
        P, flag = setNextCombination(P)
        if(flag == False): break
        arr[idx] = P
        idx += 1
        print(idx)
        print(arr)
path_choosing()

实际输出：

1
[array([0, 0, 1]), 0, 0, 0]
2
[array([0, 0, 2]), array([0, 0, 2]), 0, 0]
3
[array([0, 1, 1]), array([0, 1, 1]), array([0, 1, 1]), 0]
4
[array([0, 1, 2]), array([0, 1, 2]), array([0, 1, 2]), array([0, 1, 2])]

我的异常输出

1
[array([0, 0, 1]), 0, 0, 0]
2
[array([0, 0, 1]), array([0, 0, 2]), 0, 0]
3
[array([0, 0, 1]), array([0, 0, 2]), array([0, 1, 1]), 0]
4
[array([0, 0, 1]), array([0, 0, 2]), array([0, 1, 1]), array([0, 1, 2])]

我不知道怎么做，请帮帮我

Answer 1

P = np.zeros(n, dtype=int)

这是您在内存中创建 numpy 数组的地方。您将对该数组“P”的引用传递给 setNextCombination 方法，并对其进行更改。您实际上从未在内存中创建一个新数组，因此当您执行 arr[idx] = P 时，您每次都指向内存中的同一个数组。

您应该将 np 数组的副本传递给 setNextCombination。您可以为此使用numpy.copy。或者，在 python 中查找copy 模块。