numpy.matlib.eye#

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

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

参数:

nint: 输出的行数。
Mint, optional: 输出的列数，默认为 n。
kint, optional: 对角线的索引：0 表示主对角线，正值表示上对角线，负值表示下对角线。
dtypedtype, optional: 返回矩阵的数据类型。
order{‘C’, ‘F’}, optional: 输出是存储在内存中的行优先（C 风格）还是列优先（Fortran 风格）顺序。

返回:

I矩阵: 一个 n x M 的矩阵，其中所有元素都等于零，除了 k-th 对角线上的值为一。

另请参阅

numpy.eye: 等效的数组函数。
identity: 方形单位矩阵。

示例

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