numpy.eye

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

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

Параметры

Nint

Количество строк в выходном массиве.

Mint, optional

Количество столбцов в выходном массиве. Если None, по умолчанию равно N.

kint, optional

Индекс диагонали: 0 (по умолчанию) ссылается на главную диагональ, положительное значение — на верхнюю диагональ, отрицательное — на нижнюю диагональ.

dtypedata-type, optional

Тип данных возвращаемого массива.

order{‘C’, ‘F’}, optional

Определяет, должно ли выходное значение храниться в строчно-главной (C-стиль) или столбцово-главной (Fortran-стиль) форме в памяти.

Введено в версии 1.14.0.

likearray_like

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

Примечание

Ключевое слово like — экспериментальная функция, ожидающая принятия NEP 35.

Введено в версии 1.20.0.

Возвращает

Indarray of shape (N,M)

Массив, где все элементы равны нулю, за исключением k-й диагонали, значения которой равны единице.

См. также

identity

(почти) эквивалентная функция

diag

2-мерный массив диагонали из 1-мерного массива, указанного пользователем.

Примеры

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