如何使用Numpy.identity()

如何使用Numpy.identity()

 

Numpy.identity()的document. 输入n为行数或列数,返回一个n*n的对角阵,对角线元素为1,其余为0。dtype可选,默认为float格式。若dtype设为“bool”,则返回对角线元素为True其余元素为Flase的方阵。

a = numpy.identity(3,dtype=bool)
print(a)
[[ True False False]
[False True False]
[False False True]]