问题:希望给数组指定位置赋值,例如给3x3的全零矩阵对角线赋值为1
解决方法:
将索引用元组保存,如下所示
import numpy as np
mat = np.zeros((3,3))
x = (0, 1)
y = (0, 1)
mat[x, y] = 1
print(mat)
# array([[1., 0., 0.],
# [0., 1., 0.],
# [0., 0., 0.]])
问题:希望给数组指定位置赋值,例如给3x3的全零矩阵对角线赋值为1
解决方法:
将索引用元组保存,如下所示
import numpy as np
mat = np.zeros((3,3))
x = (0, 1)
y = (0, 1)
mat[x, y] = 1
print(mat)
# array([[1., 0., 0.],
# [0., 1., 0.],
# [0., 0., 0.]])