Is there a quick way to extract values from a numpy array?

Question

Assume there is a list that includes a numpy array such as

import numpy as np 
output = [np.array([[[5.21]],
                    [[2.22]],
                    [[1.10]],
                    [[3.76]]], dtype=np.float32)]

Is there any quick way to extract values from this output list such as

result = [5.21, 2.22, 1.10, 3.76]

many thanks

Think in steps. First get the array out of the list. Then convert the array to 1d. Finally use `tolist` to get a list. — hpaulj, Jan 11 '22 at 01:45

score 2 · Accepted Answer · answered Jan 11 '22 at 01:35

2

Yes. Try this

result = output[0].reshape(1,-1).flatten().tolist()

answered Jan 11 '22 at 01:35

Gaurav

1,095
10
19

Is there a quick way to extract values from a numpy array?

1 Answers1