Max in a sliding window in NumPy array

by

Pandas has a rolling method for both Series and DataFrames, and that could be of use here:

import pandas as pd

lst = [6,4,8,7,1,4,3,5,7,8,4,6,2,1,3,5,6,3,4,7,1,9,4,3,2]
lst1 = pd.Series(lst).rolling(5).max().dropna().tolist()

# [8.0, 8.0, 8.0, 7.0, 7.0, 8.0, 8.0, 8.0, 8.0, 8.0, 6.0, 6.0, 6.0, 6.0, 6.0, 7.0, 7.0, 9.0, 9.0, 9.0, 9.0]

For consistency, you can coerce each element of lst1 to int:

[int(x) for x in lst1]

# [8, 8, 8, 7, 7, 8, 8, 8, 8, 8, 6, 6, 6, 6, 6, 7, 7, 9, 9, 9, 9]

More Related Contents:

  1. Fitting empirical distribution to theoretical ones with Scipy (Python)?
  2. Histogram Matplotlib
  3. Link ATLAS/MKL to an installed Numpy
  4. Fastest way to grow a numpy numeric array
  5. Plotting a fast Fourier transform in Python
  6. How to check BLAS/LAPACK linkage in NumPy and SciPy?
  7. How to make scipy.interpolate give an extrapolated result beyond the input range?
  8. Inverse Distance Weighted (IDW) Interpolation with Python
  9. How to create a density plot in matplotlib?
  10. How to do n-D distance and nearest neighbor calculations on numpy arrays
  11. Relationship between SciPy and NumPy
  12. How do I install SciPy on 64 bit Windows?
  13. Multivariate spline interpolation in python/scipy?
  14. Removing duplicate columns and rows from a NumPy 2D array
  15. scipy curve_fit doesn’t like math module
  16. Numpy Resize/Rescale Image
  17. How to specify upper and lower limits when using numpy.random.normal
  18. Why is `np.sum(range(N))` very slow?
  19. Storing numpy sparse matrix in HDF5 (PyTables)
  20. How to calculate the statistics “t-test” with numpy
  21. Are NumPy’s math functions faster than Python’s?
  22. How do I use numpy.where()? What should I pass, and what does the result mean? [closed]
  23. Creating a Confidence Ellipsis in a scatterplot using matplotlib
  24. Load just part of an image in python
  25. Python/Numpy MemoryError
  26. Numpy: Divide each row by a vector element
  27. new pythonic style for shared axes square subplots in matplotlib?
  28. Numpy transpose of 1D array not giving expected result
  29. Fit mixture of two gaussian/normal distributions to a histogram from one set of data, python
  30. Speedup scipy griddata for multiple interpolations between two irregular grids

Leave a Comment