python - 根据 python 中的另一个 np.array 访问 np.array 元素

Question

我有一个numpy.ndarray (arr)，其中包含多个 3x4 数组。另一方面，我有一个 1xN referenceArray，它显示了我应该从arr.

arr=  [[1 2 3]
       [4 5 6]
       [7 8 9]
       [10 11 12]]

       [[13 14 15]
       [16 17 18]
       [19 20 21]
       [22 23 24]]

       [[25 26 27]
       [28 29 30]
       [31 32 33]
       [34 35 36]]
       .
       .
       .

referenceArray = [0 1 4 8 9]

例如

arr[0]给我子数组arr

arr:[[1 2 3]
[4 5 6]
[7 8 9]
[10 11 12]]

和arr[1]

arr:[[13 14 15]
[16 17 18]
[19 20 21]
[22 23 24]]

我的问题是，元素和大小referenceArray都是动态变化的。那么，我如何才能仅访问 中的那些子数组arr，这取决于写入的元素referenceArray并将它们附加到新数组中？

标签： pythonnumpy