Python NumPy identity方法用法及代碼示例

Numpy 的 identity(~) 方法返回指定大小的單位矩陣。

1. n | int

矩陣的大小(即行數和列數)。

2. dtype | string 或 type | optional

返回的單位矩陣所需的數據類型。默認情況下，dtype=Float 。

表示單位矩陣的 2D Numpy 數組。

創建大小為 3 的單位矩陣：

np.identity(3)



array([[1., 0., 0.],
       [0., 1., 0.],
       [0., 0., 1.]])

要創建 int64 類型的單位矩陣：

np.identity(3, dtype="int64")



array([[1, 0, 0],
       [0, 1, 0],
       [0, 0, 1]])

相關用法

注：本文由純淨天空篩選整理自Isshin Inada大神的英文原創作品 NumPy | identity method。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。