numpy.eye

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

返回一个二维数组,对角线上为 1,其他地方为 0.

参数:

Nint

输出中的行数.

M整数,可选

输出中的列数.如果为 None,则默认为 N .

k整数,可选

对角线的索引:0(默认值)表示主对角线,正值表示上对角线,负值表示下对角线.

dtypedata-type,可选

返回数组的数据类型.

order{‘C’, ‘F’},可选

输出应以行优先(C 风格)还是列优先(Fortran 风格)顺序存储在内存中.

devicestr, optional

用于放置所创建的数组的设备.默认值:None.仅用于 Array-API 互操作性,因此如果传递,则必须为 "cpu" .

在 2.0.0 版本加入.

likearray_like, optional

参考对象以允许创建非NumPy数组的数组. 如果作为 like 传入的类数组支持 __array_function__ 协议,则结果将由它定义. 在这种情况下,它可以确保创建与通过此参数传入的数组对象兼容的数组对象.

在 1.20.0 版本加入.

返回:

I形状为 (N,M) 的 ndarray

一个所有元素都等于零的数组,除了第 k 个对角线,其值等于 1.

参见

identity

(几乎)等效的函数

diag

由用户指定的 1-D 数组构成的对角线 2-D 数组.

示例

Try it in your browser!

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

Go BackOpen In Tab