numpy.identity

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

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

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

Параметры:

nint

Количество строк (и столбцов) в выходном массиве n x n.

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

Тип данных выходного массива. По умолчанию float.

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

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

Добавлена в версии 1.20.0.

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

outndarray

Массив n x n с единицами на главной диагонали и нулями во всех других элементах.

Примеры

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