上一主题

numpy.matlib.eye

下一主题

numpy.matlib.repmat

numpy.matlib.identity

numpy.matlib.identity(n, dtype=None)[source]

返回给定大小的方形单位矩阵。

参数:

n:int

返回的单位矩阵的大小。

dtype:数据类型,可选

输出的数据类型。默认为float

返回:

out:matrix

n x n矩阵,其主对角线设置为1,所有其他元素为零。

也可以看看

numpy.identity
等效数组功能。
matlib.eye
更一般的矩阵身份函数。

例子

>>> import numpy.matlib
>>> np.matlib.identity(3, dtype=int)
matrix([[1, 0, 0],
        [0, 1, 0],
        [0, 0, 1]])