numpy.identity

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

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

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

Параметры

nint

Количество строк (и столбцов) в n x n результирующей матрице.

dtypedata-type, optional

Тип данных результирующей матрицы. По умолчанию float.

likearray_like

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

Примечание

Ключевое слово like — это экспериментальная функция, ожидающая принятия NEP 35.

Введено в версии 1.20.0.

Возвращает

outndarray

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

Примеры

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