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

  统计/机器学习 Python    浏览次数:4864        分享
0

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

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

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

ccc225   2018-05-20 04:27



   1个回答 
3

对角线上元素的和就是线性代数里的迹(trace),numpy里有trace这个函数

np.trace(array)


SofaSofa数据科学社区DS面试题库 DS面经

木子周   2018-05-20 08:32



  相关主题

怎么获得numpy array每行最小值的index?   1回答

用np.round的时候报错AttributeError: no attribute 'rint'   1回答

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

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

numpy中获得tensor的维数?   2回答

numpy里的np.newaxis有什么用?   1回答

怎么把numpy array中的inf换成0?   1回答

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

numpy生成二维随机均匀分布?   1回答

怎么对2维的numpy array取整?   2回答

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

对numpy array求每行的均值   1回答



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

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

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

我们谢绝答非所问。

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

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