numpy.eye #

麻木的。eye ( N , M=None , k=0 , dtype=<class 'float'> , order='C' , * , like=None ) [来源] #

返回一个二维数组，对角线上有 1，其他位置有 0。

参数：
整数​

输出中的行数。

M整数，可选

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

k整数，可选

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

dtype数据类型，可选

返回数组的数据类型。

订单{'C', 'F'}，可选

输出是否应按行优先（C 风格）或列优先（Fortran 风格）顺序存储在内存中。

1.14.0 版本中的新增功能。

类似array_like，可选

允许创建非 NumPy 数组的引用对象。如果传入的类似数组like支持__array_function__协议，则结果将由它定义。在这种情况下，它确保创建一个与通过此参数传入的数组对象兼容的数组对象。

1.20.0 版本中的新增功能。

返回：
形状为 (N,M) 的数组

一个数组，其中所有元素都等于 0，但第k个对角线除外，其值等于 1。

也可以看看

identity

（几乎）等效函数

diag

来自用户指定的一维数组的对角二维数组。

例子

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