Reshape data after boolean indexes filtering

Question

I have a data set called DATA which regroup several 3D tables from N=173 files of individual shape (4, 4, 64) so at the end the numpy