如何创建给定一个函数返回尺寸的数组n

Question

随着numpy的和python3我不得不以下问题维数n + 1的数组：

我有返回的固定整数的2维阵列的功能大小（在这种情况下是2x3）。什么是最常用的方式来运行这个功能n次，并将它们叠加到一个3维2x3xn数组中？性能如何？只有最小数量的分配才会很好。

Answer 1

您可能正在寻找np.dstack：

>>> 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)