Python numpy trace用法及代码示例

本文简要介绍 python 语言中 numpy.trace 的用法。

用法:

numpy.trace(a, offset=0, axis1=0, axis2=1, dtype=None, out=None)

返回沿数组对角线的总和。

如果a是二维的，返回给定偏移量沿对角线的和，即元素的和a[i,i+offset]对于所有我。

如果 a 具有两个以上的维度，则由 axis1 和 axis2 指定的轴用于确定返回其轨迹的二维子数组。结果数组的形状与移除了axis1和axis2的a的形状相同。

参数：

a： array_like

输入数组，从中获取对角线。

offset： 整数，可选

对角线与主对角线的偏移量。可以是正面的也可以是负面的。默认为 0。

axis1, axis2： 整数，可选

用作对角线的二维子阵列的第一和第二轴的轴。默认值是 a 的前两个轴。

dtype： dtype，可选

确定返回数组和元素相加的累加器的数据类型。如果 dtype 的值为 None 并且 a 是精度小于默认整数精度的整数类型，则使用默认整数精度。否则，精度与 a 相同。

out： ndarray，可选

放置输出的数组。它的类型被保留，它必须具有正确的形状来保存输出。

返回：

sum_along_diagonals： ndarray

如果 a 是二维的，则返回沿对角线的和。如果 a 具有更大的维度，则返回沿对角线的和数组。

例子：

>>> 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)