【发布时间】:2010-12-28 10:14:51
【问题描述】:
假设我有三张矩阵 a、b 和 c 的“表格”,每张都有相同的 mnp 维度。我想将它们结合起来得到一个新的 mnp*3 矩阵,其 (i,j,k) 元素是 (a[i,j,k],b[i,j,k ],c[i,j,k])。我应该使用哪个命令? dstack 命令在这里似乎不起作用。谢谢。
【问题讨论】:
假设我有三张矩阵 a、b 和 c 的“表格”,每张都有相同的 mnp 维度。我想将它们结合起来得到一个新的 mnp*3 矩阵,其 (i,j,k) 元素是 (a[i,j,k],b[i,j,k ],c[i,j,k])。我应该使用哪个命令? dstack 命令在这里似乎不起作用。谢谢。
【问题讨论】:
另一个班轮是:
result = numpy.array( (a,b,c) ).transpose( (1,2,3,0) )
或更自我描述的方法:
result = empty( (m,n,p,3) )
result[:,:,:,0] = a
result[:,:,:,1] = b
result[:,:,:,2] = c
【讨论】:
我想你想要的是:
np.concatenate([np.expand_dims(x, -1) for x in (a, b, c)], axis=3)
【讨论】: