np.eye()

Единичная матрица: единицы на диагонали, нули вне её.

Сигнатураnp.eye(N, M=None, k=0)

Возвращает квадратную (или прямоугольную) матрицу с единицами на главной диагонали. Параметр k смещает диагональ.

import numpy as np
print(np.eye(3))   # [[1. 0. 0.]
                   #  [0. 1. 0.]
                   #  [0. 0. 1.]]