numpy.eye

原文：https://docs.scipy.org/doc/numpy/reference/generated/numpy.eye.html

译者：飞龙 UsyiyiCN

校对：（虚位以待）

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

返回一个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.]])