numpy.identity
numpy.identity(n, dtype=None)
Функция identity() возвращает квадратный массив с единицами по главной диагонали, при этом остальные элементы выходного массива равны 0. Эквивалентна eye(n, dtype=dtype).
-
- n - целое число
- Задает размеры квадратного массива (n x n).
- dtype - тип данных NumPy (необязательный)
- Определяет тип данных выходного массива.
-
- результат - массив NumPy
- Квадратный массив с единицами по главной диагонали, остальные элементы которого равны 0.
Примеры
>>> 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)
