我有一个数组,如下所示:
array([ [[[0.08467145], [0.0846905 ]]], [[[0.08470057], [0.08483638]]], [[[0.0846846 ], [0.08471105]]], [[[0.08469571], [0.08472978]]]], dtype=float32)
我想从每一对中提取第一个元素并存储在一个列表中,同时提取第二个元素并存储在另一个列表中。我该怎么做?
可以将数组索引与 np.ndarray.flatten :
np.ndarray.flatten
print(a[:,:,0].flatten()) print(a[:,:,1].flatten())
这将产生:
[0.08467145 0.08470057 0.0846846 0.08469571] [0.0846905 0.08483638 0.08471105 0.08472978]