numpy中的identity函数

identity函数用于一个n*n的单位矩阵(主对角线元素全为1，其余全为0的矩阵)。

np.identity(n, dtype=float）
# n:单位矩阵的边长
# dtype:可选参数,返回的数组内数据的数据类型,默认是float

>>> import numpy as np
>>> np.identity(3)
array([[1., 0., 0.],
       [0., 1., 0.],
       [0., 0., 1.]])
>>> np.identity(3, dtype='int')
array([[1, 0, 0],
       [0, 1, 0],
       [0, 0, 1]])