numpy.trace#

numpy.trace(a, offset=0, axis1=0, axis2=1, dtype=None, out=None)[来源]#

返回数组对角线上的元素之和。

如果 a 是 2-D 的，则返回沿给定偏移量的对角线之和，即元素 a[i,i+offset] 的总和（对所有 i）。

如果 a 的维度超过两个，则由 axis1 和 axis2 指定的轴用于确定要返回其迹的 2-D 子数组。结果数组的形状与 a 的形状相同，但 axis1 和 axis2 被移除。

参数:

a类数组对象: 输入数组，从中提取对角线。
offsetint, optional: 对角线相对于主对角线的偏移量。可以是正数或负数。默认为 0。
axis1, axis2int, optional: 用作 2-D 子数组的第一个和第二个轴，从中提取对角线。默认值为 a 的前两个轴。
dtypedtype, optional: 确定返回数组的数据类型以及累加器中元素的求和方式。如果 dtype 的值为 None 且 a 的整数类型精度低于默认整数精度，则使用默认整数精度。否则，精度与 a 的精度相同。
outndarray，可选: 将输出放置其中的数组。其类型将被保留，并且必须具有适合容纳输出的正确形状。

返回:

sum_along_diagonalsndarray: 如果 a 是 2-D 的，则返回沿对角线的和。如果 a 的维度更大，则返回一个沿对角线的和的数组。

另请参阅

示例

>>> import numpy as np
>>> np.trace(np.eye(3))
3.0
>>> a = np.arange(8).reshape((2,2,2))
>>> np.trace(a)
array([6, 8])

>>> a = np.arange(24).reshape((2,2,2,3))
>>> np.trace(a).shape
(2, 3)