Identify groups of continuous numbers in a list

by

EDIT 2: To answer the OP new requirement

ranges = []
for key, group in groupby(enumerate(data), lambda (index, item): index - item):
    group = map(itemgetter(1), group)
    if len(group) > 1:
        ranges.append(xrange(group[0], group[-1]))
    else:
        ranges.append(group[0])

Output:

[xrange(2, 5), xrange(12, 17), 20]

You can replace xrange with range or any other custom class.

Python docs have a very neat recipe for this:

from operator import itemgetter
from itertools import groupby
data = [2, 3, 4, 5, 12, 13, 14, 15, 16, 17]
for k, g in groupby(enumerate(data), lambda (i,x):i-x):
    print(map(itemgetter(1), g))

Output:

[2, 3, 4, 5]
[12, 13, 14, 15, 16, 17]

If you want to get the exact same output, you can do this:

ranges = []
for k, g in groupby(enumerate(data), lambda (i,x):i-x):
    group = map(itemgetter(1), g)
    ranges.append((group[0], group[-1]))

output:

[(2, 5), (12, 17)]

EDIT: The example is already explained in the documentation but maybe I should explain it more:

The key to the solution is
differencing with a range so that
consecutive numbers all appear in same
group.

If the data was: [2, 3, 4, 5, 12, 13, 14, 15, 16, 17]
Then groupby(enumerate(data), lambda (i,x):i-x) is equivalent of the following:

groupby(
    [(0, 2), (1, 3), (2, 4), (3, 5), (4, 12),
    (5, 13), (6, 14), (7, 15), (8, 16), (9, 17)],
    lambda (i,x):i-x
)

The lambda function subtracts the element index from the element value. So when you apply the lambda on each item. You’ll get the following keys for groupby:

[-2, -2, -2, -2, -8, -8, -8, -8, -8, -8]

groupby groups elements by equal key value, so the first 4 elements will be grouped together and so forth.

I hope this makes it more readable.

python 3 version may be helpful for beginners

import the libraries required first

from itertools import groupby
from operator import itemgetter

ranges =[]

for k,g in groupby(enumerate(data),lambda x:x[0]-x[1]):
    group = (map(itemgetter(1),g))
    group = list(map(int,group))
    ranges.append((group[0],group[-1]))

More Related Contents:

  1. Deleting certain elements in a list [closed]
  2. Search list for number in range
  3. Python 3 turn range to a list
  4. converting a list of integers into range in python
  5. Python range() and zip() object type
  6. Print a list in reverse order with range()?
  7. Python 3 range Vs Python 2 range
  8. Python IndexError: list index out of range
  9. Strange result when removing item from a list while iterating over it in Python
  10. How to check if a string is a substring of items in a list of strings?
  11. How to find the cumulative sum of numbers in a list?
  12. How to split a list based on a condition?
  13. Permutations between two lists of unequal length
  14. Are tuples more efficient than lists in Python?
  15. What makes sets faster than lists?
  16. Check if two unordered lists are equal [duplicate]
  17. In Python, what is the fastest algorithm for removing duplicates from a list so that all elements are unique *while preserving order*? [duplicate]
  18. Python: How to remove empty lists from a list? [duplicate]
  19. How to use python numpy.savetxt to write strings and float number to an ASCII file?
  20. How to get the n next values of a generator in a list (python)
  21. python: most elegant way to intersperse a list with an element
  22. Converting a string that represents a list, into an actual list object
  23. Merging/adding lists in Python
  24. how to keep elements of a list based on another list [duplicate]
  25. Flatten an irregular (arbitrarily nested) list of lists
  26. How to concatenate (join) items in a list to a single string
  27. “TypeError: ‘type’ object is not subscriptable” in a function signature
  28. Ellipsis lists […] and concatenating a list to itself [duplicate]
  29. Filter a list of tuples based on condition
  30. How can I convert this string to list of lists? [duplicate]

Leave a Comment