我有一个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
中的书面元素,并将它们追加到新的数组中?