指數與指數數組多維數組

我已經索引元組的數組，我想用它來從多維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是衆所周知。

任何提示？

順便說一句，'LEN（a.shape）'是通過acessible'a.ndim' –

很酷，從來不知道這一點。 –

你可以的idx轉置版本轉換爲一個元組，然後指數爲量化的解決方案 -

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

2017-02-27 10:24:08 Divakar

回答