numpy.matlib.eye#

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

返回一个对角线上为 1,其他地方为 0 的矩阵.

参数:
nint

输出中的行数.

M整数,可选

输出中的列数,默认为 n .

k整数,可选

对角线的索引:0 指的是主对角线,正值指的是上对角线,负值指的是下对角线.

dtypedtype, optional

返回矩阵的数据类型.

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

输出应以行优先(C 风格)还是列优先(Fortran 风格)顺序存储在内存中.

返回:
Imatrix

一个 n x M 矩阵,其中所有元素都等于零,除了第 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.]])