How to reshape the numpy array or list [duplicate]

Question

I'm having trouble reshaping the matrix in python.

Suppose I have four (5,5,3) matrix

a = [[[0,0,0], [0,0,0], [0,0,0], [0,0,0], [0,0,0]],
     [[0,0,0], [0,0,0], [0,0,0], [0,0,0], [0,0,0]],
     [[0,0,0], [0,0,0], [0,0,0], [0,0,0], [0,0,0]],
     [[0,0,0], [0,0,0], [0,0,0], [0,0,0], [0,0,0]],
     [[0,0,0], [0,0,0], [0,0,0], [0,0,0], [0,0,0]]]

b = [[[1, 1, 1], [1, 1, 1], [1, 1, 1], [1, 1, 1], [1, 1, 1]],
     [[1, 1, 1], [1, 1, 1], [1, 1, 1], [1, 1, 1], [1, 1, 1]],
     [[1, 1, 1], [1, 1, 1], [1, 1, 1], [1, 1, 1], [1, 1, 1]],
     [[1, 1, 1], [1, 1, 1], [1, 1, 1], [1, 1, 1], [1, 1, 1]],
     [[1, 1, 1], [1, 1, 1], [1, 1, 1], [1, 1, 1], [1, 1, 1]]]

c = [[[2, 2, 2], [2, 2, 2], [2, 2, 2], [2, 2, 2], [2, 2, 2]],
     [[2, 2, 2], [2, 2, 2], [2, 2, 2], [2, 2, 2], [2, 2, 2]],
     [[2, 2, 2], [2, 2, 2], [2, 2, 2], [2, 2, 2], [2, 2, 2]],
     [[2, 2, 2], [2, 2, 2], [2, 2, 2], [2, 2, 2], [2, 2, 2]],
     [[2, 2, 2], [2, 2, 2], [2, 2, 2], [2, 2, 2], [2, 2, 2]],]

d = [[[3, 3, 3], [3, 3, 3], [3, 3, 3], [3, 3, 3], [3, 3, 3]],
     [[3, 3, 3], [3, 3, 3], [3, 3, 3], [3, 3, 3], [3, 3, 3]],
     [[3, 3, 3], [3, 3, 3], [3, 3, 3], [3, 3, 3], [3, 3, 3]],
     [[3, 3, 3], [3, 3, 3], [3, 3, 3], [3, 3, 3], [3, 3, 3]],
     [[3, 3, 3], [3, 3, 3], [3, 3, 3], [3, 3, 3], [3, 3, 3]],]

I want to put this matrix together in the form of (5,5,4,3). like

result = [[[[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]], [[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]], [[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]], [[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]], [[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]]],
          [[[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]], [[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]], [[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]], [[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]], [[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]]],
          [[[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]], [[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]], [[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]], [[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]], [[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]]],
          [[[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]], [[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]], [[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]], [[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]], [[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]]],
          [[[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]], [[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]], [[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]], [[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]], [[0,0,0], [1, 1, 1], [2, 2, 2], [3, 3, 3]]]]

I would appreciate it if you could let me know if you have any good ideas for the solution.

Stef · Accepted Answer · 2022-06-23 06:50:51Z

0

You can stack them as numpy arrays and then convert back to list:

stacked = np.stack([a,b,c,d], axis=2).tolist()

Verify with your result: np.array_equal(stacked, result) return True.

answered Jun 23, 2022 at 6:50

Stef

30.8k3 gold badges34 silver badges60 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How to reshape the numpy array or list [duplicate]

1 Answer 1

Comments

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Linked

Related