如何使用变量来处理多维数组的元素？

Question

import numpy as np 
cube = np.zeros((2,2,2,2,2,2,2)) # Make 7 dim hypercube 
vector=[1,0,1,1,0,1,1] 
cube[vector[0],vector[1],vector[2],vector[3],vector[4],vector[5],vector[6]] # access the field [1,0,1,1,0,1,1] 

我必须使用一些高维数组，我想通过变量访问它们的字段。 上面的代码显示了一个非常糟糕的解决方案，我相信有更高效的解决方案，类似于cube[vector]或cube[vector[i] for i in range(len(vector))]，但它们都不起作用。

Answer 1

NumPy的指标只是元组，所以您可以：

cube[tuple(vector)]