How to split a numpy array in fixed size chunks with and without overlap?

by

There’s a builtin in scikit-image as view_as_windows for doing exactly that –

from skimage.util.shape import view_as_windows

view_as_windows(arr, (2,2))

Sample run –

In [40]: arr
Out[40]: 
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])

In [41]: view_as_windows(arr, (2,2))
Out[41]: 
array([[[[0, 1],
         [3, 4]],

        [[1, 2],
         [4, 5]]],


       [[[3, 4],
         [6, 7]],

        [[4, 5],
         [7, 8]]]])

For the second part, use its cousin from the same family/module view_as_blocks –

from skimage.util.shape import view_as_blocks

view_as_blocks(arr, (2,2))

More Related Contents:

  1. Using numpy to build an array of all combinations of two arrays
  2. Understanding NumPy’s einsum
  3. Intuition and idea behind reshaping 4D array to 2D array in NumPy
  4. Why is numpy’s einsum faster than numpy’s built in functions?
  5. Calculate mean across dimension in a 2D array
  6. Selecting specific rows and columns from NumPy array
  7. Convert a 1D array to a 2D array in numpy
  8. Subsetting a 2D numpy array
  9. Check if two 3D numpy arrays contain overlapping 2D arrays
  10. Selecting Random Windows from Multidimensional Numpy Array Rows
  11. How to find the index of a value in 2d array in Python?
  12. What is the difference between ndarray and array in NumPy?
  13. Python: how to store a numpy multidimensional array in PyTables?
  14. Access n-th dimension in python [duplicate]
  15. How does the axis parameter from NumPy work?
  16. Mapping a NumPy array in place
  17. How to convert a 3d numpy array to 2d
  18. Convert Python sequence to NumPy array, filling missing values
  19. numpy matrix vector multiplication [duplicate]
  20. how does multiplication differ for NumPy Matrix vs Array classes?
  21. How to split/partition a dataset into training and test datasets for, e.g., cross validation?
  22. np arrays being immutable – “assignment destination is read-only”
  23. Numpy individual element access slower than for lists
  24. How to turn a video into numpy array?
  25. Concatenate range arrays given start, stop numbers in a vectorized way – NumPy
  26. Numpy – Replace a number with NaN
  27. Why is B = numpy.dot(A,x) so much slower looping through doing B[i,:,:] = numpy.dot(A[i,:,:],x) )?
  28. numpy arange: how to make “precise” array of floats?
  29. Shift rows of a numpy array independently
  30. Formatting floats in a numpy array [duplicate]

Leave a Comment