numpy.matlib.identity

matlib.identity(n, dtype=None)

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

参数:

nint

返回的单位矩阵的大小。

dtype数据类型，可选

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

返回:

out矩阵

一个 n x n 的矩阵，其主对角线设置为 1，所有其他元素设置为 0。

另请参阅

numpy.identity

等效的数组函数。

matlib.eye

更通用的矩阵单位函数。

示例

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