怎么求一个nxn的numpy array的对角线的和？
我想到的就是跑循环，有直接求对角线和的函数么
sum = 0
for i in len(array):
sum = sum + array[i, i]
1个回答
对角线上元素的和就是线性代数里的迹（trace），numpy里有trace这个函数
np.trace(array)
木子周
2018-05-20 08:32
相关讨论
随便看看