numpy里生成单位矩阵？

np.identity(n)可以生成$n\times n$的单位矩阵

>>> np.identity(4)
array([[ 1.,  0.,  0.,  0.],
       [ 0.,  1.,  0.,  0.],
       [ 0.,  0.,  1.,  0.],
       [ 0.,  0.,  0.,  1.]])

或者也可以用np.diag生成对角阵

>>> np.diag([1] * 4)
array([[1, 0, 0, 0],
       [0, 1, 0, 0],
       [0, 0, 1, 0],
       [0, 0, 0, 1]])

最省事的是numpy.eye了，比numpy.identity还少打了几个字母，哈哈

np.eye(3)

array([[1., 0., 0.],
       [0., 1., 0.],
       [0., 0., 1.]])