Numpy使用矩阵索引插入矩阵值

import numpy as np

matrix = np.zeros((3, 3))  # 创建一个3x3的零矩阵
matrix[1, 1] = 5  # 在索引为(1, 1)的位置插入值5
print(matrix)

[[0. 0. 0.]
 [0. 5. 0.]
 [0. 0. 0.]]

import numpy as np

matrix = np.zeros((3, 3))  # 创建一个3x3的零矩阵
indices = np.array([[0, 1], [1, 2]])  # 创建一个包含插入位置的索引数组
values = np.array([1, 2])  # 创建一个包含要插入的值的数组
matrix[indices[:, 0], indices[:, 1]] = values  # 在指定位置插入对应的值
print(matrix)

[[0. 1. 0.]
 [0. 0. 2.]
 [0. 0. 0.]]

import numpy as np

matrix = np.zeros((3, 3))  # 创建一个3x3的零矩阵
slice = np.s_[1:3, 1:3]  # 创建一个切片对象，表示要插入的位置范围
matrix[slice] = 3  # 在指定的切片范围内插入值3
print(matrix)

[[0. 0. 0.]
 [0. 3. 3.]
 [0. 3. 3.]]