Диагональные массивы
Функция
diag(V, k=0) позволяет извлекать диагональ из массива, а также строить диагональные массивы из одномерных массивов.
V - подобный массиву объект, двумерные или одномерные массивы, матрицы, списки или кортежи, а также любая функция или объект с методом, возвращающие список или кортеж.
k - индекс диагонали (необязательный).
По умолчанию
k = 0, что соответствует главной диагонали. Положительное значение
k смещает диагональ вверх, отрицательное - вниз.
Пример создания диагонального массива:
V = np.diag([1, 4, 5])
print(V)
Массив
V будет иметь вид
[[1 0 0]
[0 4 0]
[0 0 5]]
Пример получения диагонального массива из матрицы
arr = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
print(np.diag(arr)) # [1 5 9]
Функция возвращает массив
NumPy (
ndarray) - указанную диагональ массива или диагональный массив из указанного одномерного массива.