numpy.eye

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

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

Параметры

Nint

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

Mint, optional

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

kint, optional

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

dtypedata-type, optional

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

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

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

New in version 1.14.0.

Возвращает

Indarray of shape (N,M)

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

См. также

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