nhận được đường chéo đối diện của mảng numpy

Question

Vì vậy, trong mảng có nhiều mảng có chức năng được xây dựng để nhận chỉ báo chéo, nhưng tôi không thể tìm ra cách để có được đường chéo bắt đầu từ trên cùng bên phải thay vì trên cùng bên trái .

Đây là mã bình thường để có được bắt đầu từ phía trên bên trái:

>>> 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]) 

vì vậy những gì để tôi sử dụng nếu tôi muốn nó trở lại:

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

Answer 1

Có

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

hoặc

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

Trong hai, np.diag(np.fliplr(array)) là nhanh hơn:

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

Dưới đây là hai ý tưởng:

step = len(array) - 1 

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

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