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指定的轴来确定返回跟踪的二维子数组。结果数组的形状与 a 具有 axis1 和 axis2 远离的。

参数

aarray_like

从中提取对角线的输入数组。

offset可选的

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

AXIS1，AXIS2可选的

用作二维子数组的第一和第二个轴的轴，从中应取对角线。默认值是的前两个轴 a .

dtype可选类型

确定返回数组和元素求和的累加器的数据类型。如果dtype具有值none和 a 是小于默认整数精度的整数类型，则使用默认整数精度。否则，精度与 a .

outndarray，可选

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

返回

sum_along_diagonals恩达雷

如果 a 是2-d，返回沿对角线的和。如果 a 具有较大的维数，然后返回沿对角线的和数组。

参见

diag, diagonal, diagflat

实例

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

numpy.linalg.slogdet numpy.linalg.solve