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-style) или столбцов (Fortran-style) в памяти.

Введено в версии 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.]])