Splitting 3d array in numpy to make smaller equal 3d array

Question

I have an array shape =(64, 64,64) I want to divide it in 8 blocks like the image, each array shape would be then (32,32,32). How do I achieve this?

amiola · Accepted Answer · 2021-12-14 12:08:51Z

1

You might reshape your array as follows:

import numpy as np
a = np.random.randn(64, 64, 64)
b = a.reshape((-1, 32, 32, 32))

You'll then see b.shape being equal to (8, 32, 32, 32).

answered Dec 14, 2021 at 12:08

amiola

3,1571 gold badge16 silver badges31 bronze badges

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

Comments

some_name.py · Accepted Answer · 2021-12-14 10:33:54Z

0

Check out numpy´s slicing:

import numpy as np

x = np.random.rand(64,64,64)

x_block_1 = x[32:64,32:64,32:64]
x_block_2 = x[0:32,32:64,32:64]
...
x_block_2 = x[32:64,0:32,0:32]

This is according to your definition.

answered Dec 14, 2021 at 10:33

some_name.py

8377 silver badges17 bronze badges