python - 我无法在 python 中正确地将我想要的值分配给变量

Question

我想操纵矩阵“b”，但由于我已将其初始值分配为等于矩阵“a”的初始值，因此矩阵“a”也被操纵，这是不希望的。我怎么解决这个问题？

import numpy as np
a=np.zeros((3,3))
b=a
b[0,:]=1
print('a=',a,'\n')

print('b=',b)

这段代码的结果是这样的：

a= [[1. 1. 1.]
 [0. 0. 0.]
 [0. 0. 0.]] 

b= [[1. 1. 1.]
 [0. 0. 0.]
 [0. 0. 0.]]

如您所见，这两个矩阵都已修改。我写了这个小代码来说明我的问题。我的主程序是一个更大的程序，但我已经认识到这个问题，就是这样。

标签： pythonnumpyvariablesvariable-assignmentassignment-operator

Answer 1

该行b = a应该是b = a.copy().

import numpy as np

a = np.zeros((3,3))
b = a.copy()
b[0,:] = 1
print('a = ',a,'\n')
print('b = ',b)
a = [[0. 0. 0.]
 [0. 0. 0.]
 [0. 0. 0.]] 
b = [[1. 1. 1.]
 [0. 0. 0.]
 [0. 0. 0.]]