python 对角线 - 实验3 简单的计算及输入输出

在Python中，实验3通常涉及到简单的计算，尤其是处理矩阵或者二维数组时，会用到对角线的概念。对角线可以分为主对角线（从左上角到右下角）和副对角线（从左下角到右上角）。下面是一个简单的例子，如何读取用户输入的矩阵，然后提取并计算主对角线和副对角线的元素之和：

# 导入必要的数学库，虽然这一步不是必需的，但它可以让计算变得更方便
import numpy as np

# 用户输入矩阵的维度（例如行数和列数）
rows = int(input("请输入矩阵的行数: "))
cols = int(input("请输入矩阵的列数: "))

# 创建空的二维列表来存储矩阵
matrix = []

# 循环获取用户输入的矩阵元素
for i in range(rows):
    row = input(f"请输入第{i+1}行的元素 (以逗号分隔): ").split(',')
    matrix.append([int(j) for j in row])

# 计算主对角线和副对角线的元素之和
main_diagonal_sum = sum(matrix[i][i] for i in range(rows))
secondary_diagonal_sum = sum(matrix[i][~i] for i in range(cols)) if cols % 2 == 0 else sum(matrix[i][~i + 1] for i in range(cols // 2))

# 输出结果
print(f"主对角线元素之和: {main_diagonal_sum}")
print(f"副对角线元素之和(如果存在): {secondary_diagonal_sum}")