python - numpy python 上的反向对角线

Question

假设我有这个:(numpy 数组)

a=
[0  1  2  3],
[4  5  6  7],
[8  9 10  11]

要得到 [1,1] 即 5 其对角线为零；根据 numpy，a.diagonal(0)= [0,5,10]。如何获得 [1,1] 的反向或从右到左对角线 [2,5,8]？这可能吗？我原来的问题是 8 x 8 (0:7)..希望对您有所帮助

Answer 1

每行反转得到一个新数组。

>>> import numpy as np
>>> a = np.array([
...     [0, 1, 2, 3],
...     [4, 5, 6, 7],
...     [8, 9, 10, 11]
... ])
>>> a[:, ::-1]
array([[ 3,  2,  1,  0],
       [ 7,  6,  5,  4],
       [11, 10,  9,  8]])
>>> a[:, ::-1].diagonal(1)
array([2, 5, 8])

或使用 numpy.fliplr :

>>> np.fliplr(a).diagonal(1)
array([2, 5, 8])