NumPy教程-numpy.matlib.identity()

这个函数用于返回一个给定尺寸的单位矩阵。单位矩阵是对角线元素初始化为1，其他元素为零的矩阵。

语法

numpy.matlib.ones(size, dtype)

参数

它接受以下参数。

1. shape: 生成的单位矩阵的行数和列数。
2. dtype: 单位矩阵的数据类型。

返回值

它返回一个指定尺寸和指定数据类型的单位矩阵。

示例

import numpy as np    
    
import numpy.matlib    
    
print(numpy.matlib.identity(4))    

输出:

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

示例：具有整数值的单位矩阵

import numpy as np    
    
import numpy.matlib    
    
print(numpy.matlib.identity(4,int))    

输出:

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