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

  统计/机器学习 Python    浏览次数:8079        分享
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



  相关讨论

python里sparse matrix怎么转成正常的numpy array显示出来?

如何查看当前numpy版本,升级numpy?

'numpy.ndarray' object has no attribute 'head'

怎么随机打乱一个numpy array?

怎么生成一个填满逻辑真(True)的numpy array?

numpy.full这个函数有什么用?

怎么对2维的numpy array取整?

numpy里的无穷大np.inf到底是多大呢?

python numpy里array和matrix的区别?

怎么把torch.tensor转成numpy ndarray?

  随便看看

deep learning中的pooling是什么意思?

为什么样本方差是除以n-1

机器学习中的奥卡姆剃刀原理是什么意思

yolo v4和yolo v3的主要区别是什么?

为什么梯度的反方向是函数下降最快的方向?