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