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 版本加入.

返回:

Imatrix

一个 n x M 矩阵,其中所有元素都等于零,除了第 k 条对角线,其值等于一.

参见

numpy.eye

等效的数组函数.

identity

方阵单位矩阵.

示例

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