numpy.eye

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

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

Параметры

Nint

Количество строк в выходном массиве.

Mint, необязательно

Количество столбцов в выходном массиве. Если None, по умолчанию N.

kint, необязательно

Индекс диагонали: 0 (по умолчанию) относится к главной диагонали, положительное значение — к верхней диагонали, отрицательное — к нижней диагонали.

dtypeтип данных, необязательно

Тип данных возвращаемого массива.

order{‘C’, ‘F’}, необязательно

Определяет, должно ли хранение выходных данных быть в строчном (C-стиле) или столбцовом (Fortran-стиле) порядке в памяти.

Введено в версии 1.14.0.

likearray_like

Ссылка на объект, позволяющая создавать массивы, которые не являются массивами NumPy. Если переданный объект like поддерживает протокол __array_function__, результат будет определяться им. В этом случае гарантируется создание объекта массива, совместимого с объектом, переданным через этот аргумент.

Введено в версии 1.20.0.

Возвращает

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