Python列表推导式

哈喽，大家好呀，下面这道例题是用列表推导式来计算矩阵的对角线之和，咱们一起学习！

例题1：用户输入一个整数n，系统自动产生n行n列的方阵，其中每个元素都是区间[1,100)内的随机整数，输出矩阵本身，以及矩阵对角线元素之和。

题目代码如下：

from random import choices
n=eval(input('请输入矩阵的行和列:'))
data=[choices(range(1,100),k=n) for i in range(n)]
print(*data,sep='\n')
print(sum([data[i][i] for i in range(len(data))]))

运行的结果如下：

eg:   请输入矩阵的行和列:3
       [35, 91, 74]
       [27, 16, 48]
       [55, 51, 72]
      123

文章持续更新中，敬请期待。