numpy.identity

numpy.identity(n, dtype=None)

Функция identity() возвращает квадратный массив с единицами по главной диагонали, при этом остальные элементы выходного массива равны 0. Эквивалентна eye(n, dtype=dtype).

Параметры:
n - целое число
Задает размеры квадратного массива (n x n).
dtype - тип данных NumPy (необязательный)
Определяет тип данных выходного массива.
Возвращает:
результат - массив NumPy
Квадратный массив с единицами по главной диагонали, остальные элементы которого равны 0.
Смотрите так же: eye, diag, diagflat

Примеры

>>> import numpy as np
>>> 
>>> np.identity(4)
array([[ 1.,  0.,  0.,  0.],
       [ 0.,  1.,  0.,  0.],
       [ 0.,  0.,  1.,  0.],
       [ 0.,  0.,  0.,  1.]])
>>> 
>>> np.identity(4, dtype = np.int8)
array([[1, 0, 0, 0],
       [0, 1, 0, 0],
       [0, 0, 1, 0],
       [0, 0, 0, 1]], dtype=int8)