Python生成随机矩阵

随机生成并输出一个3行3列矩阵A，然后求出矩阵主对角线（从左上到右下）元素之和。其中，A矩阵元素为[20,50]的随机整数，随机数种子为10。

最简单的就这样了

import numpy as np
# 设置种子
np.random.seed(10)
# 生成矩阵
arr1 = np.random.randint(20, 50, size=(3, 3))
# 对角线求和
sum(arr1.diagonal())

要哪种格式：

>>> import numpy as np
>>> from random import randint
>>> nums = [randint(20,50) for _ in range(9)]
>>> trix = np.array(nums).reshape(3,3)
>>> trix
array([[49, 44, 34],
       [41, 41, 27],
       [28, 43, 37]])
>>> import pandas as pd
>>> df = pd.DataFrame(trix)
>>> df
    0   1   2
0  49  44  34
1  41  41  27
2  28  43  37