Extracting elements from a NumPy array in Python [duplicate]

Question

I have an array that looks as follows:

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

I want to extract the first element from each pair and store in a list and also extract the second element and store it in another list. How can I do this?

BrokenBenchmark · Accepted Answer · 2022-04-29 02:11:41Z

0

You can use array indexing with np.ndarray.flatten:

print(a[:,:,0].flatten())
print(a[:,:,1].flatten())

This outputs:

[0.08467145 0.08470057 0.0846846  0.08469571]
[0.0846905  0.08483638 0.08471105 0.08472978]

answered Apr 29, 2022 at 2:11

BrokenBenchmark

19.4k7 gold badges26 silver badges40 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Extracting elements from a NumPy array in Python [duplicate]

1 Answer 1

Comments

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Linked

Related