numpy.eye#

numpy.eye(N, M=None, k=0, dtype=<class 'float'>, order='C', *, device=None, like=None)[源码]#

返回一个二维数组，其中对角线上的元素为 1，其余元素为 0。

参数:

Nint: 输出的行数。
Mint, optional: 输出的列数。如果为 None，则默认为 N。
kint, optional: 对角线的索引：0（默认）表示主对角线，正值表示上对角线，负值表示下对角线。
dtype数据类型，可选: 返回数组的数据类型。
order{‘C’, ‘F’}, optional: 输出是按行优先（C 风格）还是按列优先（Fortran 风格）顺序存储在内存中。
devicestr，可选: 用于放置创建的数组的设备。默认：None。仅用于 Array-API 互操作性，因此如果传递，必须是 "cpu"。

版本 2.0.0 中新增。
likearray_like, optional: 用于创建非 NumPy 数组的引荐对象。如果传入的 like 支持 __array_function__ 协议，则结果将由它定义。在这种情况下，它确保创建与通过此参数传入的数组兼容的数组对象。

版本 1.20.0 中新增。

返回:

Indarray of shape (N,M): 一个所有元素都等于零的数组，除了 k-th 对角线上的元素等于一。

另请参阅

identity: (几乎)等效的函数
diag: 从用户指定的 1D 数组创建对角线 2D 数组。

示例

>>> import numpy as np
>>> 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.]])