numpy.eye

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

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

Параметры:

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

диагональный 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.]])