Python numpy matlib.eye用法及代碼示例

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

用法:

matlib.eye(n, M=None, k=0, dtype=<class 'float'>, order='C')

返回一個矩陣，其中對角線為 1，其他位置為 0。

參數：

n： int

輸出中的行數。

M： 整數，可選

輸出中的列數，默認為 n。

k： 整數，可選

對角線索引：0為主對角線，正值表示上對角線，負值表示下對角線。

dtype： dtype，可選

返回矩陣的數據類型。

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

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

返回：

I： 矩陣

一個 n x M 矩陣，其中所有元素都等於 0，除了 k-th 對角線，其值等於 1。

例子：

>>> import numpy.matlib
>>> np.matlib.eye(3, k=1, dtype=float)
matrix([[0.,  1.,  0.],
        [0.,  0.,  1.],
        [0.,  0.,  0.]])