How to add a column to a 2D np array python? [duplicate]

Question

I have an 2D array :

1 2 3 4 5 6
2 6 5 2 4 1
8 7 9 0 0 0
2 3 4 5 6 2

How can I add a new column of 1D array to the above array?

So that the result will look like this?

1 2 3 4 5 6 1
2 6 5 2 4 1 0
8 7 9 0 0 0 1
2 3 4 5 6 2 0

EDIT: I'm not adding a column of zeros here.

I have tried but that ain't my question. He wants to add a column of zeros so the answer is np.zeros. My question is different. — PSN
– PSN, Commented Jan 3, 2018 at 14:36
The accepted answer suggests creating array of zeros then copy over the values, you can do the same, copy over the 2d array and then the 1d array — yosukesabai
– yosukesabai, Commented Jan 3, 2018 at 14:42
@nicola Try np.concatenate((a,b.reshape(b.shape[0],1)),axis=1) This worked well — PSN
– PSN, Commented Jan 3, 2018 at 14:55

PSN · Accepted Answer · 2018-01-03 14:58:06Z

3

np.concatenate((a,b.reshape(b.shape[0],1)),axis=1)

Solved the problem

answered Jan 3, 2018 at 14:58

PSN

2,5163 gold badges32 silver badges52 bronze badges

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

Comments

Artier · Accepted Answer · 2018-01-03 15:38:43Z

1

Try this using hstack. vstack concatenate vertically and hstack concatenate horizontally

>>> a=np.arange(0,24)
>>> a=a.reshape((4,6))
>>> a
array([[ 0,  1,  2,  3,  4,  5],
       [ 6,  7,  8,  9, 10, 11],
       [12, 13, 14, 15, 16, 17],
       [18, 19, 20, 21, 22, 23]])
>>> b=np.ones((4,1))
>>> c=np.hstack((a,b))
>>> c
array([[  0.,   1.,   2.,   3.,   4.,   5.,   1.],
       [  6.,   7.,   8.,   9.,  10.,  11.,   1.],
       [ 12.,  13.,  14.,  15.,  16.,  17.,   1.],
       [ 18.,  19.,  20.,  21.,  22.,  23.,   1.]])
>>>

answered Jan 3, 2018 at 15:38

Artier

1,6832 gold badges10 silver badges25 bronze badges

Comments

Sociopath · Accepted Answer · 2018-01-03 14:43:14Z

0

append method in numpy will do the job

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

c=np.append(a,b, axis=1)
print(c)

It gives output as follows:

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

edited Jan 3, 2018 at 14:43

answered Jan 3, 2018 at 14:37

Sociopath

13.4k22 gold badges53 silver badges82 bronze badges

6 Comments

PSN Over a year ago

No unfortunately its not.

PSN Over a year ago

ValueError: all the input arrays must have same number of dimensions This is what I'm getting.

Sociopath Over a year ago

I added few more steps to the solution

PSN Over a year ago

Yea I tried that

Sociopath Over a year ago

Did it work? It's working for me atleast. Just check if you have passed 2D array in a new variable(variable b in my case)

|

Tai · Accepted Answer · 2018-01-03 17:04:14Z

0

One can also try np.insert

x = np.array([[1, 2, 3, 4, 5, 6],
              [2, 6, 5, 2, 4, 1],
              [8, 7, 9, 0, 0, 0],
              [2, 3, 4, 5, 6, 2]])
y = np.array([1, 0, 1, 0])
# inserting at the 6th column can be achieved with this
np.insert(x, 6, y, axis=1)

array([[1, 2, 3, 4, 5, 6, 1],
       [2, 6, 5, 2, 4, 1, 0],
       [8, 7, 9, 0, 0, 0, 1],
       [2, 3, 4, 5, 6, 2, 0]])

answered Jan 3, 2018 at 17:04

Tai

8,0643 gold badges31 silver badges50 bronze badges

Collectives™ on Stack Overflow

How to add a column to a 2D np array python? [duplicate]

4 Answers 4

Comments

Comments

6 Comments

Comments

Linked

Hot Network Questions

Collectives™ on Stack Overflow

4 Answers 4

Comments

Comments

6 Comments

Comments

Linked

Related