numpy.matlib.eye

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

返回一个矩阵，对角线上的元素为 1，其他元素为 0。

参数:

nint

输出矩阵的行数。

Mint, 可选

输出矩阵的列数，默认为 n。

kint, 可选

对角线的索引：0 指主对角线，正值指上对角线，负值指下对角线。

dtypedtype, 可选

返回矩阵的数据类型。

order{‘C’, ‘F’}, 可选

输出矩阵在内存中存储的顺序，可以是行主序（C 风格）或列主序（Fortran 风格）。

版本 1.14.0 中的新功能。

返回值:

I矩阵

一个 n x M 矩阵，所有元素都等于 0，除了第 k 条对角线上的元素等于 1。

另请参阅

numpy.eye

等效的数组函数。

identity

方阵单位矩阵。

示例

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