np.identity()

这个函数和之前的区别在于,这个只能创建方阵,也就是N=M

函数的原型:np.identity(n,dtype=None)

参数:n,int型表示的是输出的矩阵的行数和列数都是n

dtype:表示的是输出的类型,默认是float

返回的是nxn的主对角线为1,其余地方为0的数组      #有点像单位矩阵

案例:

import numpy as np

a=np.identity(3)
print(a)
结果:

[[1. 0. 0.]

[0. 1. 0.]

[0. 0. 1.]]

 

原文地址:http://www.cnblogs.com/bokeyuanjj/p/16852312.html

发表评论

您的电子邮箱地址不会被公开。