Python numpy identity用法及代碼示例

本文簡要介紹 python 語言中 numpy.identity 的用法。

用法:

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

返回標識數組。

標識數組是一個正方形數組，主對角線上有一個。

參數：

n： int

n x n 輸出中的行數(和列數)。

dtype： 數據類型，可選

輸出的數據類型。默認為 float 。

like： array_like

允許創建非 NumPy 數組的引用對象。如果作為 like 傳入的類似數組支持 __array_function__ 協議，則結果將由它定義。在這種情況下，它確保創建一個與通過此參數傳入的數組對象兼容的數組對象。

返回：

out： ndarray

n x n 數組，其主對角線設置為 1，所有其他元素設置為 0。

例子：

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