计算一个NumPy数组的对角线元素的总和
有时我们需要找到右上角、左上角、右下角或左下角的对角线元素之和。Numpy为我们提供了使用numpy.trace()和numpy.diagonal()方法计算不同对角线元素之和的工具。
方法1:使用numpy.trace()寻找对角线元素的和。
语法: numpy.trace(a, offset=0, axis1=0, axis2=1, dtype=None, out=None)
例子1:对于3X3的Numpy矩阵来说
输出:
例子2:对于4X4的Numpy矩阵来说
输出:
方法2:使用numpy.diagonal()寻找对角线元素之和。
语法 :
例子1:对于3X3的Numpy矩阵来说。
输出:
例子2:对于5X5的Numpy矩阵来说
输出: