NumPy numpy.matlib.eye()的使用

该函数返回一个矩阵，其中对角元素的值初始化为1，其他位置为零。

语法

numpy.matlib.eye(n, m, k, dtype)

参数

接受以下参数。

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

返回值

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

示例

import numpy as np  

import numpy.matlib  

print(numpy.matlib.eye(n=3,m=3,k=0,dtype=int))  

输出：

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

示例：初始化浮点值

import numpy as np  

import numpy.matlib  

print(numpy.matlib.eye(n=3,m=3,k=0,dtype=float))

输出：

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