numpy.ma.identity#

ma.identity(n, dtype=None) = <numpy.ma.core._convert2ma object>#

傳回單位陣列。

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

參數:
nint

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

dtypedata-type, optional

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

likearray_like, optional

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

在 1.20.0 版本中新增。

傳回值:
outMaskedArray

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

範例

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