numpy.identity

numpy.identity(n, dtype=None, *, like=None)

傳回單位矩陣。

單位矩陣是一個主對角線上元素為 1 的方陣。

參數:

n整數

輸出 n x n 陣列中的列數（與行數）。

dtype資料類型，選用

輸出的資料類型。預設為 float。

like類陣列，選用

參考物件，允許建立非 NumPy 陣列的陣列。如果傳入為 like 的類陣列支援 __array_function__ 協定，則結果將由其定義。在這種情況下，它可以確保建立與透過此參數傳入的物件相容的陣列物件。

在版本 1.20.0 中新增。

傳回值:

outndarray

主對角線上元素設為 1，所有其他元素為 0 的 n x n 陣列。

範例

>>> import numpy as np
>>> np.identity(3)
array([[1.,  0.,  0.],
       [0.,  1.,  0.],
       [0.,  0.,  1.]])