如果函数返回维数为n的数组，如何创建维数n + 1的数组

Question

使用numpy和python3我必须遵循以下问题：

我有一个函数，它返回一个固定大小的二维数组（在这种情况下为2x3）。运行此函数do { switch (optionn) { case 1: int sum = n1 + n2; System.out.println(sum); break; case 2: System.out.println(n1 + "*" + n2 + " = " + multb); break; case 3: exit(0); default: System.out.println("Sorry, please enter valid Option"); showMenu(); }// End of switch statement int optionn = showMenu();//SHOWS THE MENU AGAIN } while (optionn == 3); System.out.println("Thank you. Good Bye."); 次的最惯用方法是什么，并将它们一起堆叠到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)