求一个nxn的numpy array的对角线的和？

  统计/机器学习 Python    浏览次数： 501
  机器学习面试也能刷题？
0

怎么求一个nxn的numpy array的对角线的和？

我想到的就是跑循环，有直接求对角线和的函数么

sum = 0
for i in len(array):
    sum = sum + array[i, i]
 

ccc225   2018-05-20 04:27


   1个回答 
2

对角线上元素的和就是线性代数里的迹（trace），numpy里有trace这个函数

np.trace(array)


SofaSofa数据科学社区 DS面经 问答 实战

木子周   2018-05-20 08:32



  相关主题

如何在numpy array尾部增加一行   2回答

删除numpy.array的最后一列   2回答

删除numpy array中指定的一列   1回答

怎么用numpy生成一个三维的tensor？   1回答

'numpy.ndarray' object has no attribute 'head'   1回答

怎么把torch.tensor转成numpy ndarray？   2回答

numpy里的np.clip函数怎么用？   1回答

python numpy里array和matrix的区别？   1回答

numpy里的无穷大np.inf到底是多大呢？   2回答

二维numpy.array转为一维的numpy.array或者list   2回答

调用np.average()时报错TypeError: No loop matching   1回答

如何往numpy array中插入一列数组？   4回答

回答问题时需要注意什么?

我们谢绝在回答前讲“生动”的故事。

我们谢绝“这么简单，你自己想”、“书上有的，你认真看”这类的回答；如果你认为对方的提问方式或者内容不妥，你可以直接忽略该问题，不用进行任何作答，甚至可以对该问题投反对票。

我们谢绝答非所问。

我们谢绝自己不会、硬要回答。

我们感激每一个用户在编写答案时的努力与付出！