从 Python 中的 NumPy 数组中提取元素

Question

我有一个如下所示的数组：

array([
      [[[0.08467145],
         [0.0846905 ]]],
       [[[0.08470057],
         [0.08483638]]],
       [[[0.0846846 ],
         [0.08471105]]],
       [[[0.08469571],
         [0.08472978]]]], dtype=float32)

我想从每一对中提取第一个元素并存储在一个列表中，同时提取第二个元素并将其存储在另一个列表中。我该怎么做？

Answer 1

您可以使用数组索引 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]

从 Python 中的 NumPy 数组中提取元素

Extracting elements from a NumPy array in Python

python

arrays

numpy