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