numpy.eye

numpy.eye(N, M=None, k=0, dtype=<class 'float'>, order='C', *, device=None, like=None)

返回一個在對角線上為 1，其他地方為 0 的 2-D 陣列。

參數:

Nint

輸出中的列數。

Mint, optional

輸出中的行數。如果為 None，則預設為 N。

kint, optional

對角線的索引：0 (預設值) 指的是主對角線，正值指的是上方對角線，負值指的是下方對角線。

dtypedata-type, optional

返回陣列的資料型別。

order{‘C’, ‘F’}, optional

輸出是否應以 row-major (C-style) 或 column-major (Fortran-style) 順序儲存在記憶體中。

devicestr, optional

要在其上放置建立陣列的裝置。預設值：None。僅用於 Array-API 互操作性，因此如果傳遞，則必須為 "cpu"。

版本 2.0.0 新增。

likearray_like, optional

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

版本 1.20.0 新增。

返回:

Indarray of shape (N,M)

一個陣列，其中所有元素都等於零，除了第 k 個對角線，其值等於一。

參見

identity

(幾乎) 等效函數

diag

從使用者指定的 1-D 陣列建立對角線 2-D 陣列。

範例

>>> import numpy as np
>>> np.eye(2, dtype=int)
array([[1, 0],
       [0, 1]])
>>> np.eye(3, k=1)
array([[0.,  1.,  0.],
       [0.,  0.,  1.],
       [0.,  0.,  0.]])