Convert Python Numpy array to array of single arrays

Question

I have a feeling that this is very easy but I can't quite figure out how to do it. Say I have a Numpy array

[1,2,3,4]

How do I convert this to

[[1],[2],[3],[4]]

In an easy way?

Thanks

To be precise: you're trying to convert np.array([1, 2, 3, 4]) to np.array([[1], [2], [3], [4]])? Or convert it to a traditional Python list of lists? — David Robinson
– David Robinson, Commented Jul 4, 2013 at 19:28

Akavall · Accepted Answer · 2013-07-04 19:31:43Z

3

You can use np.newaxis:

>>> a = np.array([1,2,3,4] 
array([1, 2, 3, 4])
>>> a[:,np.newaxis]
array([[1],
       [2],
       [3],
       [4]])

answered Jul 4, 2013 at 19:31

Akavall

86.8k58 gold badges214 silver badges261 bronze badges

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

Comments

Ashwini Chaudhary · Accepted Answer · 2013-07-04 19:42:16Z

2

>>> import numpy as np
>>> a = np.array([1,2,3,4])
>>> np.reshape(a, (-1, 1))
array([[1],
       [2],
       [3],
       [4]])

If you want normal python list then use list comprehension:

>>> a = np.array([1,2,3,4]) 
>>> [[x] for x in a]
[[1], [2], [3], [4]]

answered Jul 4, 2013 at 19:28

Ashwini Chaudhary

252k60 gold badges478 silver badges519 bronze badges

Pawel Miech · Accepted Answer · 2013-07-04 19:35:17Z

1

The most obvious way that comes to mind is:

>>> new = []
>>> for m in a:
        new.append([m])

but this creates normal Python's list of lists, I'm not sure if this is what you want...

answered Jul 4, 2013 at 19:35

Pawel Miech

7,8624 gold badges41 silver badges58 bronze badges

Milan · Accepted Answer · 2013-07-04 20:40:11Z

1

>>> A = [1,2,3,4]

>>> B = [[x] for x in A]

>>> print B
[[1], [2], [3], [4]]

answered Jul 4, 2013 at 19:40

Milan

3612 silver badges4 bronze badges