numpy.matlib.identity

numpy.matlib.identity(n, dtype=None)

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

参数：

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]])