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.]]