numpy.eye

numpy.eye(N, M=None, k=0, dtype=) [source]

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

Параметры:	N : int Количество строк в выводе. M : int, необязательно Количество столбцов в выводе. Если None, по умолчанию равно N. k : int, необязательно Индекс диагонали: 0 (по умолчанию) относится к главной диагонали, положительное значение к верхней диагонали, а отрицательное — к нижней диагонали. dtype : тип данных, необязательно Тип данных возвращаемого массива.
Возвращает:	I : ndarray формы (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.]])