NumPy教程-numpy.matlib.identity()
这个函数用于返回一个给定尺寸的单位矩阵。单位矩阵是对角线元素初始化为1,其他元素为零的矩阵。
语法
numpy.matlib.ones(size, dtype)
参数
它接受以下参数。
shape: 生成的单位矩阵的行数和列数。
dtype: 单位矩阵的数据类型。
返回值
它返回一个指定尺寸和指定数据类型的单位矩阵。
👉点击领取:最全Python资料合集
示例
import numpy as npimport numpy.matlibprint(numpy.matlib.identity(4))
输出:
[[1. 0. 0. 0.][0. 1. 0. 0.][0. 0. 1. 0.][0. 0. 0. 1.]]
示例:具有整数值的单位矩阵
import numpy as npimport numpy.matlibprint(numpy.matlib.identity(4,int))
输出:
[[1 0 0 0][0 1 0 0][0 0 1 0][0 0 0 1]]
热门推荐