Python numpy matlib.identity用法及代码示例

本文简要介绍 python 语言中 numpy.matlib.identity 的用法。

用法:

matlib.identity(n, dtype=None)

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

参数：

n： int

返回的单位矩阵的大小。

dtype： 数据类型，可选

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

返回：

out： 矩阵

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

例子：

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