NumPy numpy.matlib.identity()的使用

该函数用于返回给定大小的单位矩阵。单位矩阵的对角线元素初始化为1，其他所有元素初始化为0。

语法

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