numpy - 如何实现“内部”拉链？

Question

x两个数组的形状y为(a,b)。如何获得形状的组合数组(a,b,2)？

我目前的解决方案是

z = np.zeros((a,b,2))
z[:,:,0] = x
z[:,:,1] = y

是否可以在不创建新数组的情况下实现这一目标？

Answer 1

您可以使用np.dstack：

In [2]: import numpy as np
In [3]: a = np.random.normal(size=(4,6))
In [4]: b = np.random.normal(size=(4,6))

In [5]: np.dstack((a,b)).shape
Out[5]: (4, 6, 2)

和一个比较：

In [10]: d = np.dstack((a,b))  
In [11]: c = np.zeros((4,6,2))
In [12]: c[:,:,0] = a
In [13]: c[:,:,1] = b

In [14]: np.allclose(c,d)
Out[14]: True