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

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

1. a | array-like

输入数组。

2. k | int | optional

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

输入数组的对角线。

考虑以下二维数组：

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



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

获取主对角线：

np.diag(a)



array([1, 5, 9])

要使用 k=1 获取对角线：

np.diag(a, k=1)



array([2, 6])

要使用 k=-2 获取对角线：

np.diag(a, k=-2)



array([7])

相关用法

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