How to use numpy.dstack in a loop?

Question

I\'m trying to populate an array by using 2D arrays with np.dstack.

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

         


        

Answer 1

You do not need a loop. You need a list or a tuple of arrays:

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