python - 使用多个索引从 n 维数组中访问元素的优化方式

Question

我来自R背景，如果它太基本了，请原谅我的无知。我想知道是否有更好的方法来访问 N 维数组中的元素给定多个索引 inb 一次而不是循环遍历索引列表？以下是和的可重现input代码output

#  TestData
import numpy as np

np.random.seed(0)
test_input = np.random.randint(1000, size=144).reshape(1,12, 12,1)
indices = [np.array([[ 0,  1, 55,  0], 
                     [ 0,  5, 74,  0],
                     [ 0,  8, 55,  0],
                     [ 0, 14, 76,  0]])]

输出是这些索引中存在的值的列表或元组

我正在尝试类似的东西test_input[indices]，显然它不会起作用。

标签： pythonarraysnumpy