python - 如何在循环中使用 numpy.dstack？

Question

我正在尝试通过将二维数组与 np.dstack 结合使用来填充数组。

m1 = np.array([[1,1],[1,1]])
m2 = np.array([[2,2],[2,2]])
m3 = np.array([[3,3],[3,3]])

lst = m1
lst = np.dstack((lst,m2))
lst = np.dstack((lst,m3))

在循环中执行此操作的正确方法是什么？我正在寻找类似的东西

lst = np.empty(...)
for _
    lst = np.dstack((lst,variable2Darray))

Answer 1

您不需要循环。您需要一个列表或数组元组:

m1 = np.array([[1,1],[1,1]])
m2 = np.array([[2,2],[2,2]])
m3 = np.array([[3,3],[3,3]])

arrays = m1, m2, m3 # A tuple of arrays
lst = np.dstack(arrays) # Stack them all