numpy.matlib.eye()

这个函数返回一个矩阵，其中对角线元素初始化为1，其他地方为零。

语法

参数

它接受以下参数。

1. n: 它表示结果矩阵中的行数。
2. m: 它表示结果矩阵中的列数。
3. k: 它是对角线的索引。
4. dtype: 它是输出的数据类型。

返回

返回一个具有未初始化条目的矩阵。

示例

输出

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

示例：初始化浮点数值

输出

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