加入并重塑两个ndarray

Question

以下是将两个数组a和b组合在一起以生成数组c的实际示例：

import numpy as np

a = np.array([[1,2,3], [4,5,6], [7,8,9], [10,11,12]])
b = -a
c = np.reshape(np.array(zip(np.ravel(a), np.ravel(b))), (4,3,2))

print "a = \n", a, "\n"
print "b = \n", b, "\n"
print "c = \n", c

运行它会显示我想要的结果数组c：

我必须想象有一种更光滑的方式来实现这一目标。是否有更多的Pythonic方式？

Answer 1

您可以使用numpy中的dstack函数来实现您的目标：

 np.dstack((a,b)).reshape((4,3,2))

希望它有所帮助！