np.identity()

二、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
版权声明：本文为博主原创文章，转载请附上博文链接！