numpy.trace_NumPy中文网

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子数组。所得数组的形状与除去了轴1 和轴2的a的形状相同。

参量

一个array_like: 输入数组，从中获取对角线。
offset int，可选: 对角线与主对角线的偏移量。既可以是正面的，也可以是负面的。预设为0。
axis1，axis2 int，可选: 用作2-D子阵列的第一轴和第二轴的轴，应该从中获取对角线。缺省值是前两个轴线一个。
dtype dtype，可选: 确定将元素求和的返回数组和累加器的数据类型。如果dtype的值是None且a是精度小于默认整数精度的整数类型，则使用默认整数精度。否则，精度是相同的一个。
out ndarray，可选: 输出放置到的数组。它的类型被保留，并且必须具有正确的形状以容纳输出。

退货

也可以看看

例子

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