创建一个i行j列的矩阵

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

import numpy as np
#方法一
i,j=input().split()
array1=np.random.randint(1,76,size=(int(i),int(j)))
print(array1)
print(np.sum(np.diagonal(array1)))

#方法二
array2=[[np.random.randint(1,76) for x in range(int(j))] for y in range(int(i))]
print(array2)
print(sum([array2[z][z] for z in range(min(int(i),int(j)))]))