快速解释
如果您希望 A 成为原始矩阵,您的函数应该是这样的。只需使用np.copy()
功能
def zeros(A):
new_mat = np.copy(A) #JUST COPY IT WITH np.copy() function
for i in range(len(A)):
for j in range(len(A[i])):
if A[i][j]==0:
for b in range(len(A)):
new_mat[b][j] = 0
else:
new_mat[i][j] = A[i][j]
return A
详尽的解释
假设我们不希望 numpy ndarraya
有别名。例如,a
当我们取(例如)它的一个切片时,我们希望防止更改它的任何值,我们将这个切片分配给b
然后我们修改 b 的一个元素。我们要避免这种情况:
import numpy as np
a = np.array([[1,2,3],[4,5,6],[7,8,9]])
b = a[0]
b[2] = 50
a
Out[]:
array([[ 1, 2, 50],
[ 4, 5, 6],
[ 7, 8, 9]])
现在您可能认为将 numpy-ndarray 对象视为列表对象可能会解决我们的问题。但它也不起作用:
%reset #delete all previous variables
a = np.array([[1,2,3],[4,5,6],[7,8,9]])
b = a[:][0] #this is how you would have copied a slice of "a" if it was a list
b[2] = 50
a
Out[]:
array([[ 1, 2, 50], #problem persists
[ 4, 5, 6],
[ 7, 8, 9]])
在这种情况下,如果您将 numpy 数组视为与 python 列表不同的对象,问题就会得到解决。为了复制 numpy 数组,您不能这样做copy = name_array_to_be_copied[:]
,但是copy = np.copy(name_array_to_be_copied)
. 因此,这将解决我们的别名问题:
%reset #delete all previous variables
import numpy as np
a = np.array([[1,2,3],[4,5,6],[7,8,9]])
b = np.copy(a)[0] #this is how you copy numpy ndarrays. Calling np.copy() function
b[2] = 50
a
Out[]:
array([[1, 2, 3], #problem solved
[4, 5, 6],
[7, 8, 9]])
PS 注意这个zeros()
功能。即使在修复了别名问题之后,您的函数也不会将 new_matrix 中的列转换为 0,这些列在矩阵 A 的同一列中至少有一个零(这是我认为您希望通过查看错误报告的输出来完成的你的函数[[1, 0, 3], [5, 0, 78], [7, 0, 45]]
,因为它实际上产生[[1,0,3],[5,0,78],[7,3,45]]
)。如果你愿意,你可以试试这个:
def zeros_2(A):
new_mat = np.copy(A)
for i in range(len(A[0])): #I assume each row has same length.
if 0 in new_mat[:,i]:
new_mat[:,i] = 0
print(new_mat)
return A