numpy.eye

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

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

Параметры:

Nint

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

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

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

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

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

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

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

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

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

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

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

Устройство, на котором нужно разместить созданный массив. По умолчанию: None. Только для совместимости с Array-API, поэтому должен быть "cpu" при передаче.

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

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

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

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

Возвращаемое значение:

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