numpy.matlib.identity()

此函数用于返回给定大小的单位矩阵。单位矩阵是指对角线元素初始化为 1，所有其他元素初始化为零的矩阵。

语法

参数

它接受以下参数。

1. shape：它是生成的单位矩阵中的行数和列数。
2. dtype：它是单位矩阵的数据类型。

返回

它返回指定大小和指定数据类型的单位矩阵。

示例

输出

[[1. 0. 0. 0.]
 [0. 1. 0. 0.]
 [0. 0. 1. 0.]
 [0. 0. 0. 1.]]

示例：具有整数值的单位矩阵

输出

[[1 0 0 0]
 [0 1 0 0]
 [0 0 1 0]
 [0 0 0 1]]