Combining an array using Python and NumPy

Question

I have two arrays of the form:

a = np.array([1,2,3])
b = np.array([4,5,6])

Is there a NumPy function which I can apply to these arrays to get the followng output?

[[1,4],[2,5][3,6]]

eumiro · Accepted Answer · 2012-04-11 10:37:24Z

7

np.vstack((a,b)).T

returns

array([[1, 4],
       [2, 5],
       [3, 6]])

and

np.vstack((a,b)).T.tolist()

returns exactly what you need:

[[1, 4], [2, 5], [3, 6]]

answered Apr 11, 2012 at 10:37

eumiro

214k36 gold badges307 silver badges264 bronze badges

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

2 Comments

I am hoping to keep it as a numpy array. Is that possible?

Yes, np.vstack((a,b)).T returns a numpy array.