小编Mol*_*usq的帖子

我有一个任意维度的Numpy数组，以及一个索引向量，其中每个维度包含一个数字。我想获得与索引集相对应的数组切片，小于所有维度的索引数组中的值，例如

A = np.array([[1, 2, 3, 4],
              [5, 6, 7, 8],
              [9,10,11,12]])
index = [2,3]

result = [[1,2,3],
          [5,6,7]]

直观的语法类似于A[:index]，但这由于明显的原因而无法使用。

如果数组的尺寸是固定的，我可以写A[:index[0],:index[1],... :index[n]]; 我可以使用某种列表理解A[:i for i in index]吗？