给定一个返回n维数组的函数，如何创建n+1维数组

Question

对于numpy和python3，我必须解决以下问题：

我有一个函数，它返回一个固定大小的整数的二维数组（本例中是2x3）。运行这个函数n次并将它们叠加到一个3维2x3xn数组中，最惯用的方法是什么？表现如何？一些只做最小分配数的东西会很好。你知道吗

Answer 1

您可能正在寻找^{}：

>>> import numpy as np
>>> arrs = [np.random.rand(2, 3) for x in range(5)]
>>> np.dstack(arrs).shape
(2, 3, 5)

如果您知道最终形状，可以执行以下操作：

>>> out = np.empty((2, 3, 5))
>>> out[..., 0] = np.random.rand(2, 3)