二、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.]]
---------------------
作者：爱抠脚的coder
来源：CSDN
原文：https://blog.csdn.net/m0_37393514/article/details/81455915
版权声明：本文为博主原创文章，转载请附上博文链接！