Add dimension numpy array

Question

I have two numpy arrays such as:

a = np.array([1, 2, 3])

b = np.array([101, 102, 103 ])

I want to create a new array with shape (len(a), 2), such as

array([[1, 101], [2, 102], [3, 103]])

How can I do it with numpy?

BENY · Accepted Answer · 2019-03-25 14:34:38Z

2

This is so called column_stack

np.column_stack((a,b))
Out[309]: 
array([[  1, 101],
       [  2, 102],
       [  3, 103]])

answered Mar 25, 2019 at 14:34

BENY

324k22 gold badges176 silver badges250 bronze badges

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

1 Comment

@Alex yw :-) happy coding :-)

Learning is a mess · Accepted Answer · 2019-03-25 15:59:45Z

1

Alternatively:

np.c_[a,b]

will do the job as well.

answered Mar 25, 2019 at 15:59

8,3178 gold badges42 silver badges82 bronze badges