CODEWITHSUNDEEP
pythonarraysnumpymultidimensional-arraynumpy-ndarray
Damien Rice
Damien Rice

Reputation: 101

How to combine two arrays of different dimension in Python?

I do have two arrays which are (150,1400,1200) and (150,1400). I want to know how to combine these two arrays row and column wise in Python?

Upvotes: 1

Views: 167

Answers (2)

Hrusikesh
Hrusikesh

Reputation: 661

Hey you can have a look at the below link to resolve your issue

Concat two arrays of different dimensions numpy

Upvotes: 0

I'mahdi
I'mahdi

Reputation: 24049

IIUC, You need to add one dimension to b then use numpy.concatenate

a = np.random.rand(150, 1400, 1200)
print('a.shape:',a.shape)
b = np.random.rand(150, 1400)
print('b.shape:',b.shape)
print('b[..., None].shape:',b[..., None].shape)

c = np.concatenate([a, b[..., None]], axis=2)
print('c.shape:',c.shape)

Output:

a.shape: (150, 1400, 1200)
b.shape: (150, 1400)
b[..., None].shape: (150, 1400, 1)
c.shape: (150, 1400, 1201)

Upvotes: 3

Related Questions