numpy.identity

numpy.identity(n, dtype=None) [source]

Возвращает единичную матрицу.

Единичная матрица — это квадратная матрица с единицами на главной диагонали.

Параметры:	n : int Количество строк (и столбцов) в выходном массиве n x n. dtype : data-type, optional Тип данных выходного массива. По умолчанию float.
Возвращает:	out : ndarray Массив n x n, где главная диагональ состоит из единиц, а все остальные элементы равны 0.

Примеры

>>> np.identity(3)
array([[1.,  0.,  0.],
       [0.,  1.,  0.],
       [0.,  0.,  1.]])