reshaping array into a square array python,"x.reshape(2, 2, 5).transpose(1, 0, 2)"