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

2024-05-19 20:12:15 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个numpy.ndarray (arr),它包含多个3x4数组。另一方面,我有一个1xNreferenceArray,它显示了我应该从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中写入的元素,并将它们附加到新数组中


Tags: numpy元素数组ndarrayarrreferencearray