怎么求一个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
相关主题
怎么把torch.tensor转成numpy ndarray？
2回答
能直接把csv格式的数据文件读到numpy array里吗？
1回答
'numpy.ndarray' object has no attribute 'head'
1回答
调用np.average()时报错TypeError: No loop matching
1回答
我们谢绝在回答前讲“生动”的故事。
我们谢绝“这么简单，你自己想”、“书上有的，你认真看”这类的回答；如果你认为对方的提问方式或者内容不妥，你可以直接忽略该问题，不用进行任何作答，甚至可以对该问题投反对票。
我们谢绝答非所问。
我们谢绝自己不会、硬要回答。
我们感激每一个用户在编写答案时的努力与付出！