python关于矩阵的问题有人会吗

输入两个正整数i和j，然后创建一个i行j列的矩阵（包含子列表的列表），其中每个元素都是区间[1,76）内的随机整数，最后输出这个矩阵和对角线元素之和

import numpy as np
import random as r

i, j = list(map(int, input('输入两个正整数（用空格隔开）：').split()))

matrix = np.array(r.sample(range(1,76),i*j)).reshape(i,j)

print(matrix) #测试用，可去掉

print(matrix.diagonal()) #测试用，可去掉

print(sum(matrix.diagonal())) #输出结果

测试结果：
输入两个正整数（用空格隔开）：5 5
[[60 55 49 28 50]
[ 4 75 47 18 37]
[22 7 15 41 3]
[44 5 9 72 10]
[74 8 1 67 34]]
[60 75 15 72 34]
256