reshaping of array in numpy