Я хочу взять некоторые индексы из элементов numpy.array. Это то, что у меня так далеко:
import numpy as np
a = np.array(range(20))
a.shape=4,5
#~ [[ 0 1 2 3 4]
#~ [ 5 6 7 8 9]
#~ [10 11 12 13 14]
#~ [15 16 17 18 19]]
idxs = np.array([(4,0),(3,1),(2,1),(0,3)])
ret = a.take(idxs,1).diagonal().transpose()
print ret
#~ [[ 4. 0.]
#~ [ 8. 6.]
#~ [ 12. 11.]
#~ [ 15. 18.]]
Возможно ли получить этот результат более простым (или более быстрым) способом?