numpy.ma.identity

numpy.ma.identity(n, dtype=None) = <numpy.ma.core._convert2ma object>

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

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

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

Примеры

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