numpy.eye

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

返回一个2-D数组，其中一个在对角线上，零在其他地方。

参数：

N：int

输出中的行数。

M：int，可选

输出中的列数。如果为None，默认为N。

k：int，可选

对角线索引：0（默认）是指主对角线，正值是指上对角线，负值是指下对角线。

dtype：数据类型，可选

返回数组的数据类型。

返回：

I：形状的数组（N，M）

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

也可以看看

identity

（几乎）等效函数

diag

对角2-D数组由用户指定的1-D数组组成。

例子

>>> 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.]])