numpy trace 用法及代码示例

Lincoln Numpy 函数 4年前 0 541

用法：
numpy.trace(a, offset=0, axis1=0, axis2=1, dtype=None, out=None)
返回沿数组对角线的和。

如果a为2-D，则返回其对角线与给定偏移量之和，即元素之和a[i,i+offset]为了我

如果a具有两个以上的维，则使用axis1和axis2指定的轴来确定要返回其迹线的2-D子数组。所得数组的形状与除去了轴1和轴2的a的形状相同。

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

–

描述

参数

a：： array_like

输入数组，从中获取对角线。
offset：： int, 可选参数

对角线与主对角线的偏移量。既可以是正面的，也可以是负面的。默认为0。
axis1, axis2：： int, 可选参数

用作2-D子阵列的第一轴和第二轴的轴，应该从中获取对角线。默认值为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)

声明：本站所有文章，如无特殊说明或标注，均为本站原创发布。任何个人或组织，在未征得本站同意时，禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益，可联系我们进行处理。