numpy.eye

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

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

Параметры:

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

Возвращает:

I : ndarray формы (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.]])