How do I reshape a NumPy multi dimensional array to another array with the same dimensions, but different shape?

Question

I have a NumPy multi dimensional array of shape (1,76,76,255). I want to reshape it to another multidimensional array of shape (1,255,76,76). It’s still a 4D array, but I need to change the data indices I guess.

Is there an easy way without using loops?

@hpaulj np.transpose will give you arr.shape[::-1] not the one OP wants — tbhaxor
– tbhaxor, Commented Mar 16, 2022 at 2:28

tbhaxor · Accepted Answer · 2022-03-16 02:27:31Z

1

The function you are looking for is np.moveaxis() which lets you move a source axis to its destination.

>>> arr = np.random.random((1,76,76,255))
>>> 
>>> arr.shape
(1, 76, 76, 255)
>>> arr2 = np.moveaxis(arr, 3, 1)
>>> arr2.shape
(1, 255, 76, 76)
>>>

Please note that these axes are 0-indexed

answered Mar 16, 2022 at 2:27

tbhaxor

2,0572 gold badges17 silver badges52 bronze badges

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

2 Comments

hpaulj Over a year ago

transpose can do anything moveaxs can. arr.tranpose(0,3,1,2)

tbhaxor Over a year ago

I didn't know that. Thanks for mentioning @hpaulj

Collectives™ on Stack Overflow

How do I reshape a NumPy multi dimensional array to another array with the same dimensions, but different shape?

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related