numpy.diagonal#

numpy.diagonal(a, offset=0, axis1=0, axis2=1)[源代码]#

返回指定的对角线。

如果 a 是 2-D 的，则返回具有给定偏移量的 a 的对角线，即形式为 a[i, i+offset] 的元素集合。如果 a 的维度大于 2，则使用 axis1 和 axis2 指定的轴来确定返回其对角线的 2-D 子数组。结果数组的形状可以通过移除 axis1 和 axis2 并附加一个等于结果对角线大小的索引到右侧来确定。

在 NumPy 1.7 之前的版本中，此函数始终返回一个包含对角线值副本的独立新数组。

在 NumPy 1.7 和 1.8 中，它继续返回对角线的副本，但依赖于此事实已被弃用。写入结果数组将继续按原样工作，但会发出一个 FutureWarning。

从 NumPy 1.9 开始，它返回原始数组的只读视图。尝试写入结果数组将产生一个错误。

在未来的某个版本中，它将返回一个读/写视图，写入返回的数组将修改您的原始数组。返回的数组将具有与输入数组相同的类型。

如果您不写入此函数返回的数组，则可以忽略以上所有内容。

如果您依赖当前行为，我们建议显式复制返回的数组，即使用 np.diagonal(a).copy() 而不是 np.diagonal(a)。这对于过去和未来的 NumPy 版本都将起作用。

参数:
a类数组对象

从中提取对角线的数组。

offsetint, optional

对角线相对于主对角线的偏移量。可以是正数或负数。默认为主对角线 (0)。

axis1int, optional

用作从中提取对角线的 2-D 子数组的第一个轴。默认为第一个轴 (0)。

axis2int, optional

用作从中提取对角线的 2-D 子数组的第二个轴。默认为第二个轴 (1)。

返回:
array_of_diagonalsndarray

如果 a 是 2-D 的，则返回一个包含对角线的 1-D 数组，其类型与 a 相同，除非 a 是 matrix，在这种情况下，为了保持向后兼容性，将返回一个 1-D 数组而不是 (2-D) matrix。

如果 a.ndim > 2，则移除由 axis1 和 axis2 指定的维度，并在末尾插入一个新轴，该轴对应于对角线。

引发:
ValueError

如果 a 的维度小于 2。

另请参阅

diag

MATLAB 风格的 1-D 和 2-D 数组操作。

diagflat

创建对角线数组。

trace

沿对角线求和。

示例

>>> import numpy as np
>>> a = np.arange(4).reshape(2,2)
>>> a
array([[0, 1],
       [2, 3]])
>>> a.diagonal()
array([0, 3])
>>> a.diagonal(1)
array([1])

一个 3-D 示例

>>> a = np.arange(8).reshape(2,2,2); a
array([[[0, 1],
        [2, 3]],
       [[4, 5],
        [6, 7]]])
>>> a.diagonal(0,  # Main diagonals of two arrays created by skipping
...            0,  # across the outer(left)-most axis last and
...            1)  # the "middle" (row) axis first.
array([[0, 6],
       [1, 7]])

我们刚刚获取了其主对角线的子数组；请注意，每个子数组对应于固定最右边的（列）轴，并且对角线被“打包”在行中。

>>> a[:,:,0]  # main diagonal is [0 6]
array([[0, 2],
       [4, 6]])
>>> a[:,:,1]  # main diagonal is [1 7]
array([[1, 3],
       [5, 7]])

通过使用 numpy.flipud 或 numpy.fliplr 反转元素顺序可以获得反斜对角线。

>>> a = np.arange(9).reshape(3, 3)
>>> a
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])
>>> np.fliplr(a).diagonal()  # Horizontal flip
array([2, 4, 6])
>>> np.flipud(a).diagonal()  # Vertical flip
array([6, 4, 2])

请注意，检索对角线的顺序因翻转函数而异。

在此页面