Как найти сумму главной диагонали матрицы в Python? 🧮
Для того чтобы посчитать сумму главной диагонали матрицы в Python, вы можете использовать следующий код:
matrix = [[1, 2, 3],
[4, 5, 6],
[7, 8, 9]]
diagonal_sum = sum(matrix[i][i] for i in range(len(matrix)))
print(diagonal_sum)
В этом коде мы создаем матрицу, а затем с помощью генератора списков и функции sum суммируем элементы главной диагонали.
Детальный ответ
Как найти сумму главной диагонали матрицы в Python?
Для нахождения суммы элементов главной диагонали матрицы в Python, мы можем использовать цикл для перебора элементов этой диагонали и накапливать их сумму. Давайте рассмотрим подробнее.
1. Создание матрицы
Прежде чем мы начнем находить сумму главной диагонали, нам необходимо создать матрицу. Матрица - это двумерный массив элементов. В Python мы можем представить его с помощью списка списков. Ниже приведен пример создания матрицы с использованием встроенной функции input()
:
matrix = []
size = int(input("Введите размер матрицы: "))
for i in range(size):
row = []
for j in range(size):
element = int(input(f"Введите элемент {i+1},{j+1}: "))
row.append(element)
matrix.append(row)
В приведенном выше коде мы сначала создаем пустой список matrix
. Затем мы запрашиваем у пользователя размер матрицы и используем вложенный цикл for
для ввода элементов в каждую строку. Мы преобразуем каждый ввод в целое число с помощью int()
и добавляем элемент в текущую строку. После завершения каждой строки мы добавляем ее в список matrix
.
2. Нахождение суммы главной диагонали
Теперь, когда у нас есть матрица, давайте перейдем к нахождению суммы ее главной диагонали. Главная диагональ - это линия, проходящая через элементы с индексами (i, i), где i - это индекс строки и столбца.
sum_diagonal = 0
for i in range(size):
sum_diagonal += matrix[i][i]
В этом коде мы инициализируем переменную sum_diagonal
с нулевым значением, которая будет использоваться для накопления суммы элементов главной диагонали. Затем мы используем цикл for
, чтобы перебирать значения от 0 до размера матрицы. На каждой итерации мы добавляем текущий элемент главной диагонали к sum_diagonal
. Индексы для доступа к элементам главной диагонали равны (i, i), поэтому мы используем matrix[i][i]
.
3. Вывод суммы главной диагонали
Наконец, мы можем вывести найденную сумму главной диагонали, чтобы пользователь мог увидеть результат. Давайте сделаем это:
print(f"Сумма главной диагонали: {sum_diagonal}")
В этом коде мы используем функцию print()
, чтобы вывести фразу "Сумма главной диагонали:" и значение переменной sum_diagonal
.
Полный пример кода
matrix = []
size = int(input("Введите размер матрицы: "))
for i in range(size):
row = []
for j in range(size):
element = int(input(f"Введите элемент {i+1},{j+1}: "))
row.append(element)
matrix.append(row)
sum_diagonal = 0
for i in range(size):
sum_diagonal += matrix[i][i]
print(f"Сумма главной диагонали: {sum_diagonal}")
Вот и все! Теперь вы знаете, как найти сумму главной диагонали матрицы в Python. Не забудьте, что главная диагональ проходит через элементы с одинаковыми индексами (i, i), где i - индекс строки и столбца. Путем использования цикла for
и переменной для накопления суммы вы можете легко решить эту задачу.