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