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