numpy.eye

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

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

Параметры:

N : int Количество строк в выходном массиве. M : int, optional Количество столбцов в выходном массиве. Если None, по умолчанию равно N. k : int, optional Индекс диагонали: 0 (по умолчанию) соответствует главной диагонали, положительное значение — выше главной диагонали, отрицательное — ниже главной диагонали. dtype : data-type, optional Тип данных возвращаемого массива. order : {‘C’, ‘F’}, optional Определяет, хранится ли выходной массив в порядке строк (C-стиль) или столбцов (Fortran-стиль) в памяти. Новое в версии 1.14.0.

Возвращает:

I : ndarray 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.]])