numpy.eye()函数

参考文章:https://docs.scipy.org/doc/numpy/reference/generated/numpy.eye.html
numpy.eye（N，M =无，k = 0，dtype = <class’flove’>，order =‘C’ ）
返回一个二维数组，其中对角线为1，零点为零。

参数：
N ： int
输出中的行数。
M ： int，可选
输出中的列数。如果无，默认为Ñ。
k ： int，可选
对角线的索引：0（默认值）指的是主对角线，正值指的是上对角线，负值指的是下对角线。
dtype ： 数据类型，可选
返回数组的数据类型。
order： {‘C’，‘F’}，可选
输出是否应以内存中的行主（C风格）或列主（Fortran风格）顺序存储。

返回：
一个数组，其中所有元素都等于零，除了第k个对角线，其值等于1。

import numpy as np
print(np.eye(2, dtype=int))
输出结果：[[1 0]
          [0 1]]
print(np.eye(3, k=1))   
输出结果：[[0. 1. 0.]
          [0. 0. 1.]
          [0. 0. 0.]]