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

Numpy 的 eye(~) 方法返回一個 2D 數組，其對角線填充 1，所有其他條目填充 0。

1. N | int

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

2. M | int | optional

所需的列數。默認情況下，M=N。

3. k | int | optional

對角線的偏移量。如果為正，則對角線將向上移動，否則對角線將向下移動。默認情況下，k=0 。

4. dtype | string 或 type | optional

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

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

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

np.eye(3)



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

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

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



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

要創建具有 4 列的數組，請按如下方式設置 M=4：

np.eye(3, M=4)



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

要將對角線向上移動 1：

np.eye(3, k=1)



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

要將對角線向下移動 2：

np.eye(3, k=-2)



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

相關用法

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