numpy.eye() 生成对角矩阵

numpy.eye(N,M=None, k=0, dtype=<type 'float'>)

关注第一个第三个参数就行了

第一个参数：输出方阵（行数=列数）的规模，即行数或列数

第三个参数：默认情况下输出的是对角线全“1”，其余全“0”的方阵，如果k为正整数，则在右上方第k条对角线全“1”其余全“0”，k为负整数则在左下方第k条对角线全“1”其余全“0”。

>>> np.eye(2, dtype=int)  
array([[1, 0],  
       [0, 1]])  
>>> np.eye(3, k=1)  
array([[ 0.,  1.,  0.],  
       [ 0.,  0.,  1.],  
       [ 0.,  0.,  0.]])