关于“将数组堆叠到数组”的Python编码问题

Question

我有一个关于“数组内堆叠数组”的简单 Python 编码问题。在这里，我有几个数组如下：

a1 = np.array([[1,1,1],[1,1,1],[1,1,1]]) 
a2 = np.array([[2,2,2],[2,2,2],[2,2,2]])
a3 = np.array([[3,3,3],[3,3,3],[3,3,3]])
a4 = np.array([[4,4,4],[4,4,4],[4,4,4]])
a5 = np.array([[5,5,5],[5,5,5],[5,5,5]])

而且，这是我想要的输出：。我想将a1到a5数组堆叠成一个数组。

但是，我没有使用 for 循环迭代生成此类输出的想法。您能提供一些解决这个问题的想法吗？

谢谢！

Answer 1

毫不奇怪，要堆叠数组，您可以使用 stack 函数：

np.stack([a1, a2, a3, a4, a5])

输出：

array([[[1, 1, 1],
        [1, 1, 1],
        [1, 1, 1]],

       [[2, 2, 2],
        [2, 2, 2],
        [2, 2, 2]],

       [[3, 3, 3],
        [3, 3, 3],
        [3, 3, 3]],

       [[4, 4, 4],
        [4, 4, 4],
        [4, 4, 4]],

       [[5, 5, 5],
        [5, 5, 5],
        [5, 5, 5]]])