Python练习题 028：求3*3矩阵对角线数字之和

2023-07-31 00:36| 来源: 网络整理| 查看: 265

【Python练习题 028】求一个3*3矩阵对角线元素之和

-----------------------------------------------------

这题解倒是解出来了，但总觉得代码太啰嗦。矩阵这东西，应该有个很现成的方法可以直接计算才对…… 啰嗦代码如下：

str = input('请输入9个数字，用空格隔开，以形成3*3矩阵：') n = [int(i) for i in str.split(' ')] #获取9个数字 mx = [] #存储矩阵 for i in range(3): mt = [] #收集矩阵中每行的数字列表 for j in range(3): mt.append(n[i*3 + j]) mx.append(mt) #每行数字存储进mx列表 print(mt) sum1 = 0 sum2 = 0 for i in range(3): sum1 += mx[i][i] sum2 += mx[2-i][i] print('\方向对角线3元素之和为：%s' % sum1) print('/方向对角线3元素之和为：%s' % sum2)

输出结果如下：

请输入9个数字，用空格隔开，以形成3*3矩阵：7 9 6 5 4 8 6 7 3[7, 9, 6][5, 4, 8][6, 7, 3]\方向对角线3元素之和为：14/方向对角线3元素之和为：16

只要将代码稍微改一下，就可以接受任意 n*n 矩阵的计算了，而且数字数目输入有误时，也可以提示重新输入。改后代码如下：

import math err = True while err: str = input('请输入n**2个数字，用空格隔开，以形成n*n的矩阵：') n = [int(i) for i in str.split(' ')] #获取9个数字 t = math.floor(math.sqrt(len(n))) if t == math.sqrt(len(n)) and t != 1: err = False else: print('您输入的数字个数有误，请重试。') mx = [] #存储矩阵 for i in range(t): mt = [] #收集矩阵中每行的数字列表 for j in range(t): mt.append(n[i*t + j]) mx.append(mt) #每行数字存储进mx列表 print(mt) sum1 = 0 sum2 = 0 for i in range(t): sum1 += mx[i][i] sum2 += mx[t-1-i][i] print('\方向对角线元素之和为：%s' % sum1) print('/方向对角线元素之和为：%s' % sum2)

输出结果如下：

请输入n**2个数字，用空格隔开，以形成n*n的矩阵：8 7 5 4 3 5 6 7 4 5 3 4 2 9 8 7 5 8 9 6 5 2 3 2 6[8, 7, 5, 4, 3][5, 6, 7, 4, 5][3, 4, 2, 9, 8][7, 5, 8, 9, 6][5, 2, 3, 2, 6]\方向对角线元素之和为：31/方向对角线元素之和为：19

虽然凑合能用，但我仍然坚信，Python肯定已经准备好了什么函数或方法，可以轻松计算矩阵。容我有空好好搜索一番，再来更新。

++++++++++++++++++++++++++++++++++++++

题目出处：编程语言入门经典100例【Python版】

【本文地址】

公司简介

联系我们