How to find the groups of consecutive elements in a NumPy array

by 
def consecutive(data, stepsize=1):
    return np.split(data, np.where(np.diff(data) != stepsize)[0]+1)

a = np.array([0, 47, 48, 49, 50, 97, 98, 99])
consecutive(a)

yields

[array([0]), array([47, 48, 49, 50]), array([97, 98, 99])]

More Related Contents:

  1. Replacing values represented by 'UN' with NaN
  2. ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()
  3. How to find the exact intersection of a curve (as np.array) with y==0?
  4. How to access the ith column of a NumPy multidimensional array?
  5. pandas select from Dataframe using startswith
  6. Concatenating two one-dimensional NumPy arrays
  7. What’s the fastest way in Python to calculate cosine similarity given sparse matrix data?
  8. gradient descent using python and numpy
  9. Sorting a 2D numpy array by multiple axes
  10. Weighted standard deviation in NumPy
  11. Vectorized NumPy linspace for multiple start and stop values
  12. Pandas pd.Series.isin performance with set versus array
  13. Efficiently detect sign-changes in python
  14. Efficient way to take the minimum/maximum n values and indices from a matrix using NumPy
  15. Get week start date (Monday) from a date column in Python (pandas)?
  16. What is dtype(‘O’), in pandas?
  17. Python baseline correction library
  18. How to get faster code than numpy.dot for matrix multiplication?
  19. numpy float: 10x slower than builtin in arithmetic operations?
  20. Finding the nearest value and return the index of array in Python
  21. What is the purpose of meshgrid in Python / NumPy?
  22. Generating Discrete random variables with specified weights using SciPy or NumPy
  23. Index 2D numpy array by a 2D array of indices without loops
  24. How to overplot a line on a scatter plot in python?
  25. Python: find contour lines from matplotlib.pyplot.contour()
  26. No speedup when summing uint16 vs uint64 arrays with NumPy?
  27. Fastest way to compute entropy in Python
  28. Swapping columns in a numpy array?
  29. Compute a confidence interval from sample data assuming unknown distribution
  30. Efficient dot products of large memory-mapped arrays

Leave a Comment