How to insert a column in a multidimensional array?

Question

This seem to be a trivial question, yet I did not find the answer I am looking for. I have a 2D array say:

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

And another column

b = np.array([9,11])
bt = np.reshape(b, (2,1))

I would like to add/append the bt column at the zero column of array a. I tried using numpy.insert:

tt = np.insert(a,0,bt,axis=1)

But the result is:

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

What I want is:

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

What am I doing wrong?

Mehraban · Accepted Answer · 2015-03-16 08:30:25Z

2

You can use numpy.column_stack to do that:

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

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

answered Mar 16, 2015 at 8:30

Mehraban

3,3525 gold badges40 silver badges63 bronze badges

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

Comments

Carsten · Accepted Answer · 2015-03-16 08:30:15Z

2

You can either directly use b:

tt = np.insert(a, 0, b, axis=1)
print tt

[[ 9  1  3  5]
 [11  2  4  6]]

Or, if you are starting with something shaped like bt, transpose it:

tt = np.insert(a, 0, bt.T, axis=1)
print tt

[[ 9  1  3  5]
 [11  2  4  6]]

answered Mar 16, 2015 at 8:30

Carsten

18.5k4 gold badges51 silver badges56 bronze badges

Comments

Kasravnd · Accepted Answer · 2015-03-16 08:34:11Z

0

As an alternative alongside np.hstack you can play with indexing :

>>> c=np.zeros((a.shape[0],a.shape[1]+1))
>>> c[::,0]=b
>>> c[::,1:]=a
>>> c
array([[  9.,   1.,   3.,   5.],
       [ 11.,   2.,   4.,   6.]])

answered Mar 16, 2015 at 8:34

Kasravnd

108k19 gold badges167 silver badges195 bronze badges

2 Comments

icypy Over a year ago

This is an interesting solution. But I am not fully following. What are: >>>c[::,0]=b >>> c[::,1:]=a

Kasravnd Over a year ago

@Arcticpython its column assignment ! c[::,0]=b will assign b on the firs column of c and c[::,1:]=a will assign the a on the second column of c to end

Collectives™ on Stack Overflow

How to insert a column in a multidimensional array?

3 Answers 3

Comments

Comments

2 Comments

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

Comments

2 Comments

Related