Python NumPy diagonal方法用法及代码示例

NumPy 的 diagonal(~) 方法返回输入数组的对角线。

从 NumPy 1.9 开始，diagonal(~) 返回输入数组的只读视图。修改此视图将导致错误。如果要修改它，请使用NumPy的copy(~)方法。

NumPy 团队正在考虑在未来版本中将行为从只读更改为read-and-write。

1. a | array-like

输入数组。

2. offset | int | optional

如果 offset 为正，则将返回顶部的下一个 offset 对角线。如果为负数，则将返回底部的对角线。默认情况下，offset=0 。

3. axis1 | int | optional

要从中提取对角线的第一个轴。默认情况下，轴1=0。

4. axis2 | int | optional

从中提取对角线的第二个轴。默认情况下，轴2=0。

仅当处理具有 3 个或更多维度的数组时，才需要考虑参数 axis1 和 axis2。

包含输入数组对角线的 NumPy 数组。

考虑以下二维数组：

a = np.array([[1,2,3],[4,5,6],[7,8,9]])
a



array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])

获取主对角线：

np.diagonal(a)



array([1, 5, 9])

获取偏移量为 1 的对角线：

np.diagonal(a, offset=1)



array([2, 6])

要获得偏移量为 -2 的对角线：

np.diagonal(a, offset=-2)



array([7])

相关用法

注：本文由纯净天空筛选整理自Isshin Inada大神的英文原创作品 NumPy | diagonal method。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。