使用 Python 建立單位矩陣

線上性代數中，n 維單位矩陣是一個 n × n 方陣，主對角線上為 1，其他處為 0。本文將解釋如何使用 Python 程式語言的 NumPy 庫建立單位矩陣。

使用 Python 建立單位矩陣

NumPy 是一個 Python 程式語言庫，用於建立大型多維陣列和矩陣。使用 pip3 install numpy 命令安裝 NumPy 庫以建立單位矩陣。

這個庫的 identity() 函式將一個數字作為引數，並返回一個具有該行數和列數的標識陣列。你可以選擇指定輸出的資料型別。它預設使用 float。

在下面的示例中建立了一個具有 integer 資料型別的 4x4 單位矩陣。

import numpy as np

matrix_int = np.identity(4, dtype = int)
print(matrix_int)

如果未指定 dtype 引數，則使用 float 資料型別建立矩陣。

import numpy as np

matrix_float = np.identity(4)
print(matrix_float)