Python numpy eye用法及代碼示例

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

用法:

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

返回一個二維數組，其中對角線為 1，其他位置為 0。

參數：

N： int

輸出中的行數。

M： 整數，可選

輸出中的列數。如果沒有，則默認為 N。

k： 整數，可選

對角線索引：0(默認)表示主對角線，正值表示上對角線，負值表示下對角線。

dtype： 數據類型，可選

返回數組的數據類型。

order： {‘C’, ‘F’}，可選

輸出是否應按行優先(C 風格)或列優先(Fortran-style)順序存儲在內存中。

like： array_like

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

返回：

I： 形狀的ndarray(N，M)

一個數組，其中所有元素都等於 0，除了 k-th 對角線，其值等於 1。

例子：

>>> 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.]])