numpy.eye#

numpy.eye(N, M=None, k=0, dtype= 'float'>, order='C', *, device=None, like=None)[источник]#

Возвращает 2-D массив с единицами на диагонали и нулями в остальных местах.

Параметры:

Nint: Количество строк в выводе.
Mint, необязательный: Количество столбцов в выводе. Если None, по умолчанию N.
kint, необязательный: Индекс диагонали: 0 (по умолчанию) относится к главной диагонали, положительное значение относится к верхней диагонали, а отрицательное значение — к нижней диагонали.
dtypeтип данных, опционально: Тип данных возвращаемого массива.
порядок{'C', 'F'}, опционально: Должен ли вывод храниться в порядке row-major (C-стиль) или column-major (Fortran-стиль) в памяти.
devicestr, optional: Устройство, на котором размещается созданный массив. По умолчанию: None. Только для совместимости с Array-API, поэтому должно быть "cpu" если передано.

Новое в версии 2.0.0.
какarray_like, необязательный: Объект-ссылка, позволяющий создавать массивы, которые не являются массивами NumPy. Если массивоподобный объект, переданный как like поддерживает __array_function__ протокол, результат будет определен им. В этом случае он гарантирует создание объекта массива, совместимого с переданным через этот аргумент.

Новое в версии 1.20.0.

Возвращает:

Яndarray формы (N,M): Массив, в котором все элементы равны нулю, за исключением k-я диагональ, значения которой равны единице.

Смотрите также

identity: (почти) эквивалентная функция
diag: диагональный 2-D массив из 1-D массива, указанного пользователем.

Примеры

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