Python numpy matrix eye()用法及代码示例

numpy.matlib.eye()是另一个用于在numpy中进行矩阵运算的函数。它返回一个矩阵，对角线上有一个，其他位置为零。

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

参数：
n :[int]输出矩阵中的行数。
M :[int，可选]输出矩阵中的列数，默认为n。
k :[int，可选]对角线的索引。 0表示主对角线，正值表示上对角线，负值表示下对角线，默认为0。
dtype :[可选]所需的输出数据类型。
order:在内存中是以行优先(C-style)还是列主要(Fortran-style)顺序存储多维数据。

Return :一个n x M矩阵，其中所有元素均等于0，除了k-th对角线(其值等于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]]

相关用法

注：本文由纯净天空筛选整理自jana_sayantan大神的英文原创作品 numpy matrix operations | eye() function。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。