Simple conversion of the 1D array in python

Question

i have a huge dataset as array in the format below

 #input array
    array([ 5.94911 , -1.0366  ,  3.25678 ])

i need to convert it to arrays in the format below.

array([[5.94911],
       [-1.0366],
       [ 3.25678]])

Grigory Feldman · Accepted Answer · 2020-01-13 08:32:04Z

4

a = array([ 5.94911 , -1.0366  ,  3.25678 ])
dataset = np.array(a).reshape(-1,1)

This solution will not use additional memory for reshaping.

answered Jan 13, 2020 at 3:28

Grigory Feldman

4133 silver badges7 bronze badges

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

Comments

U13-Forward · Accepted Answer · 2020-01-13 03:22:58Z

2

>>> a = array([ 5.94911 , -1.0366  ,  3.25678 ])
>>> np.vstack(a)
array([[ 5.94911],
       [-1.0366 ],
       [ 3.25678]])
>>>

Or use a list comprehension:

>>> a = array([ 5.94911 , -1.0366  ,  3.25678 ])
>>> np.array([[i] for i in a])
array([[ 5.94911],
       [-1.0366 ],
       [ 3.25678]])
>>>

But of course vstack is recommended.

answered Jan 13, 2020 at 3:22

U13-Forward

71.8k15 gold badges100 silver badges125 bronze badges