Функция eye
Функция eye создает двумерный массив NumPy
с единицами на главной диагонали и нулями вне её.
Первым параметром передается количество строк,
вторым - количество столбцов (по умолчанию равно
количеству строк). Третьим параметром можно задать
смещение диагонали относительно главной.
Синтаксис
np.eye(N, [M], [k], [dtype])
Параметры:
-
N- количество строк в массиве -
M- количество столбцов (опционально) -
k- смещение диагонали (опционально, по умолчанию 0) -
dtype- тип данных элементов (опционально)
Пример
Давайте создадим единичную матрицу размером 3×3:
import numpy as np
arr = np.eye(3)
print(arr)
Результат выполнения кода:
[[1. 0. 0.]
[0. 1. 0.]
[0. 0. 1.]]
Пример
Теперь создадим прямоугольную матрицу 2×4 с единицами на главной диагонали:
import numpy as np
arr = np.eye(2, 4)
print(arr)
Результат выполнения кода:
[[1. 0. 0. 0.]
[0. 1. 0. 0.]]
Пример
Создадим матрицу 3×3 со смещением диагонали на единицу вверх:
import numpy as np
arr = np.eye(3, k=1)
print(arr)
Результат выполнения кода:
[[0. 1. 0.]
[0. 0. 1.]
[0. 0. 0.]]
Пример
Создадим матрицу 4×4 со смещением диагонали на две позиции вниз:
import numpy as np
arr = np.eye(4, k=-2)
print(arr)
Результат выполнения кода:
[[0. 0. 0. 0.]
[0. 0. 0. 0.]
[1. 0. 0. 0.]
[0. 1. 0. 0.]]
Пример
Укажем тип данных элементов как целые числа:
import numpy as np
arr = np.eye(3, dtype=int)
print(arr)
Результат выполнения кода:
[[1 0 0]
[0 1 0]
[0 0 1]]
