python 对角线 - 实验3 简单的计算及输入输出
时间: 2024-10-20 07:19:27 浏览: 18
在Python中,实验3通常涉及到简单的计算,尤其是处理矩阵或者二维数组时,会用到对角线的概念。对角线可以分为主对角线(从左上角到右下角)和副对角线(从左下角到右上角)。下面是一个简单的例子,如何读取用户输入的矩阵,然后提取并计算主对角线和副对角线的元素之和:
```python
# 导入必要的数学库,虽然这一步不是必需的,但它可以让计算变得更方便
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}")
阅读全文