Как получить диагональные элементы матрицы numpy?

Чтобы получить диагональные элементы матрицы numpy, можно воспользоваться методом numpy.diag. Этот метод возвращает одномерный массив, содержащий значения диагонали.

import numpy as np

# Создаем матрицу
matrix = np.array([[1, 2, 3],
                   [4, 5, 6],
                   [7, 8, 9]])

# Получаем диагональные элементы
diagonal_elements = np.diag(matrix)

print(diagonal_elements)
# Вывод: [1 5 9]

Как получить диагональные элементы матрицы numpy

Если вам нужно получить диагональные элементы матрицы в библиотеке numpy, есть несколько способов сделать это. Ниже я расскажу вам о двух самых распространенных подходах.

1. Использование функции numpy.diagonal()

Первый способ - использование функции numpy.diagonal(). Эта функция позволяет получить диагональные элементы из двумерной матрицы.

import numpy as np

# Создаем матрицу
matrix = np.array([[1, 2, 3],
                   [4, 5, 6],
                   [7, 8, 9]])

# Получаем диагональные элементы
diagonal_elements = np.diagonal(matrix)

print(diagonal_elements)

В этом примере мы создаем матрицу размером 3x3 и используем функцию diagonal() для получения диагональных элементов. В результате выполнения кода мы получим массив, содержащий диагональные элементы: [1 5 9].

2. Использование индексации

Второй способ - использование индексации для получения диагональных элементов.

import numpy as np

# Создаем матрицу
matrix = np.array([[1, 2, 3],
                   [4, 5, 6],
                   [7, 8, 9]])

# Получаем диагональные элементы
diagonal_elements = matrix.diagonal()

print(diagonal_elements)

Здесь мы также создаем матрицу размером 3x3, но вместо использования функции numpy.diagonal() мы вызываем метод diagonal() непосредственно из объекта матрицы. Результат будет таким же - массив [1 5 9].

Теперь вы знаете два способа получения диагональных элементов матрицы в библиотеке numpy. Выберите тот, который лучше соответствует вашим потребностям и уровню удобства использования.