Splitting list based on missing numbers in a sequence

by

Python 3 version of the code from the old Python documentation:

>>> # Find runs of consecutive numbers using groupby.  The key to the solution
>>> # is differencing with a range so that consecutive numbers all appear in
>>> # same group.
>>> from itertools import groupby
>>> from operator import itemgetter
>>> data = [ 1,  4,5,6, 10, 15,16,17,18, 22, 25,26,27,28]
>>> for k, g in groupby(enumerate(data), lambda i_x: i_x[0] - i_x[1]):
...     print(list(map(itemgetter(1), g)))
...
[1]
[4, 5, 6]
[10]
[15, 16, 17, 18]
[22]
[25, 26, 27, 28]

The groupby function from the itertools module generates a break every time the key function changes its return value. The trick is that the return value is the number in the list minus the position of the element in the list. This difference changes when there is a gap in the numbers.

The itemgetter function is from the operator module, you’ll have to import this and the itertools module for this example to work.

Alternatively, as a list comprehension:

>>> [map(itemgetter(1), g) for k, g in groupby(enumerate(seq2), lambda i_x: i_x[0] - i_x[1])]
[[1, 2], [4, 5, 6], [8, 9, 10]]

More Related Contents:

  1. finding the biggest sequence of repeating numbers in a list
  2. Get number of items from list (or other iterable) with certain condition
  3. Python finding repeating sequence in list of integers?
  4. Find intersection of two nested lists?
  5. What exactly does the .join() method do?
  6. Interleave multiple lists of the same length in Python
  7. Compute list difference
  8. Convert Python sequence to NumPy array, filling missing values
  9. Why does foo.append(bar) affect all elements in a list of lists?
  10. List of dicts to/from dict of lists
  11. Transpose nested list in python
  12. Combine lists with common elements
  13. How do I split a string into a list?
  14. How do I prepend to a short python list?
  15. Python: why does my list change when I’m not actually changing it?
  16. Select 50 items from list at random
  17. Using any() and all() to check if a list contains one set of values or another
  18. Head and tail in one line
  19. Find min, max, and average of a list
  20. How to split list and pass them as separate parameter?
  21. Convert list into list of lists
  22. Big-O of list slicing
  23. Repeat each item in a list a number of times specified in another list
  24. Adding a string to a list using augmented assignment
  25. convert csv file to list of dictionaries
  26. What is the meaning of list[:] in this code? [duplicate]
  27. Is there a way to circumvent Python list.append() becoming progressively slower in a loop as the list grows?
  28. Difference between consecutive elements in list [duplicate]
  29. Python list([]) and []
  30. Python : how to append new elements in a list of list?

Leave a Comment