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 處的值。