索引具有索引數組的多維數組

Question

我有一個索引元組數組，我想用它從多維numpy數組中挑選出值，

import numpy
a = numpy.random.rand(10, 10, 10)
idx = [[1, 1, 2], [0, 3, 7], [9, 8, 4], [9, 9, 9]]

理解[a[i, j, k] for i, j, k in idx]只有在知道a.ndim時才有效。

任何提示？

Answer 1

您可以將idx的轉置版本轉換為元組，然后為矢量化解決方案編制索引 -

a[tuple(np.transpose(idx))]