python - Numpy:按最后一维堆叠数组

Question

假设我有 3 个形状相同的 numpy 数组 a、b、c

a.shape == b.shape == c.shape == (7,9)

现在我想创建一个大小为 (7,9,3) 的 3 维数组，比如 x，这样

x[:,:,0] == a
x[:,:,1] == b
x[:,:,2] == c

执行此操作的“pythonic”方式是什么(也许在一行中)？

提前致谢!

Answer 1

有一个函数可以做到这一点:numpy.dstack (“d”代表“深度”)。例如:

In [10]: import numpy as np

In [11]: a = np.ones((7, 9))

In [12]: b = a * 2

In [13]: c = a * 3

In [15]: x = np.dstack((a, b, c))

In [16]: x.shape
Out[16]: (7, 9, 3)

In [17]: (x[:, :, 0] == a).all()
Out[17]: True

In [18]: (x[:, :, 1] == b).all()
Out[18]: True

In [19]: (x[:, :, 2] == c).all()
Out[19]: True