NumPy 矩阵索引

本教程将介绍指定 NumPy 矩阵索引的方法。

NumPy 矩阵索引

数组索引用于通过在数组内指定元素的索引来访问元素。如果我们有一个用零填充的数组，并希望将特定值放在数组内的特定索引处，我们可以使用数组索引方法。对于 Python 中的一维和二维数组，数组索引的工作方式非常不同。如果我们想像处理一维数组一样访问二维数组的前两个元素，我们必须使用 Array[(0,1),(0,1)] 索引。

import numpy as np

matrix = np.zeros((3,3))

values = np.array([1,2,3])

matrix[(0,1,2),(0,1,2)] = values
print(matrix)

输出：

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

我们使用 NumPy 矩阵索引将矩阵 matrix 中特定索引处的零替换为 values 数组中的值。我们首先创建了一个矩阵 matrix 并用零填充它。然后，我们创建了数组 values，其中包含我们想要输入到矩阵中的值。然后我们使用 matrix[(0,1,2),(0,1,2)] = values 访问矩阵内的值。它替换了 matrix 的索引 0,0、1,1 和 2,2 处的值。