Python 带索引数组的索引多维数组

我有一个索引元组数组，我想用它从多维numpy数组中挑选值

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

只有在已知a.ndim的情况下，对idx中i，j，k的理解才有效

---

有什么提示吗？

您可以将转置版本的

idx

转换为元组，然后为矢量化解决方案编制索引-

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

---

顺便说一句，

len（a.shape）

可以通过

a.ndim

实现，这很酷，我从来不知道。