Uzyskanie przeciwległej przekątnej tablicy numpy

Question

Tak więc w tablicach numpy jest wbudowana funkcja uzyskiwania wskaźników diagonalnych, ale nie mogę się zorientować, jak ustawić przekątną, zaczynając od prawego górnego rogu, a nie od lewego górnego rogu .

Jest to normalny kod, aby uzyskać zaczynając od górnej lewej:

>>> import numpy as np 
>>> array = np.arange(25).reshape(5,5) 
>>> diagonal = np.diag_indices(5) 
>>> array 
array([[ 0, 1, 2, 3, 4], 
    [ 5, 6, 7, 8, 9], 
    [10, 11, 12, 13, 14], 
    [15, 16, 17, 18, 19], 
    [20, 21, 22, 23, 24]]) 
>>> array[diagonal] 
array([ 0, 6, 12, 18, 24]) 

więc co mogę używać, jeśli chcę wrócić:

array([ 4, 8, 12, 16, 20]) 

Answer 1

Jest

In [47]: np.diag(np.fliplr(array)) 
Out[47]: array([ 4, 8, 12, 16, 20]) 

lub

In [48]: np.diag(np.rot90(array)) 
Out[48]: array([ 4, 8, 12, 16, 20]) 

Z tych dwóch, np.diag(np.fliplr(array)) jest szybsze:

In [50]: %timeit np.diag(np.fliplr(array)) 
100000 loops, best of 3: 4.29 us per loop 

In [51]: %timeit np.diag(np.rot90(array)) 
100000 loops, best of 3: 6.09 us per loop 

Answer 2

Oto dwa pomysły:

step = len(array) - 1 

# This will make a copy 
array.flat[step:-step:step] 

# This will make a veiw 
array.ravel()[step:-step:step]