Combine matrix in numpy

Question

Suppose I have three "sheets" of matrix a,b and c, each with the same mnp dimension. And I want to combine them to get a new mnp*3 matrix whose (i,j,k) element is (a[i,j,k],b[i,j,k],c[i,j,k]). Which command should I use ? The dstack command seems not work here. Thanks.

unutbu · Accepted Answer · 2009-12-20 03:19:59Z

4

Another one liner would be:

result = numpy.array( (a,b,c) ).transpose( (1,2,3,0) )

or a more self-descriptive method:

result = empty( (m,n,p,3) )
result[:,:,:,0] = a
result[:,:,:,1] = b
result[:,:,:,2] = c

edited Dec 20, 2009 at 3:19

unutbu

886k197 gold badges1.9k silver badges1.7k bronze badges

answered Dec 20, 2009 at 3:03

Vincent Marchetti

4,9583 gold badges21 silver badges8 bronze badges

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

Comments

Alex Martelli · Accepted Answer · 2009-12-20 02:41:53Z

2

I think what you want is:

np.concatenate([np.expand_dims(x, -1) for x in (a, b, c)], axis=3)

answered Dec 20, 2009 at 2:41

Alex Martelli

887k175 gold badges1.3k silver badges1.4k bronze badges

Collectives™ on Stack Overflow

Combine matrix in numpy

2 Answers 2

Comments

Comments

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Related