无涯教程-numpy - matlib.eye()函数

67 阅读1分钟

此函数返回一个矩阵,其对角线元素初始化为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.]]  

参考链接

www.learnfk.com/numpy/numpy…