numpy矩阵操作eye()函数

numpy.matlib.eye()是另一个在numpy中进行矩阵运算的函数。它返回一个对角线上有1，其他地方有0的矩阵。

语法: numpy.matlib.eye(n, M=None, k=0, dtype=’float’, order=’C’)

参数 :
n : [int] 输出矩阵中的行数。
M : [int, optional] 输出矩阵的列数，默认为n。
k : [int, optional] 对角线的索引。0指的是主对角线，正值指的是上对角线，负值指的是下对角线。默认是0。
dtype : [可选] 希望输出的数据类型。
order : 是否以行为主（C风格）或列为主（Fortran风格）的顺序在内存中存储多维数据。

返回：一个n x M的矩阵，其中所有元素都等于零，除了第k条对角线的数值等于1。

代码#1：

# Python program explaining
# numpy.matlib.eye() function
  
# importing matrix library from numpy
import numpy as geek
import numpy.matlib
  
# desired 3 x 3 output matrix 
out_mat = geek.matlib.eye(3, k = 0) 
print ("Output matrix : ", out_mat) 

输出 :

Output matrix :  
[[ 1.  0.  0.]
 [ 0.  1.  0.]
 [ 0.  0.  1.]]

代码#2：

# Python program explaining
# numpy.matlib.eye() function
  
# importing numpy and matrix library
import numpy as geek
import numpy.matlib
  
# desired 4 x 5 output matrix 
out_mat = geek.matlib.eye(n = 4, M = 5, k = 1, dtype = int) 
print ("Output matrix : ", out_mat) 

输出 :

Output matrix :  
[[0 1 0 0 0]
 [0 0 1 0 0]
 [0 0 0 1 0]
 [0 0 0 0 1]]