Finding clusters of numbers in a list

by

Not strictly necessary if your list is small, but I’d probably approach this in a “stream-processing” fashion: define a generator that takes your input iterable, and yields the elements grouped into runs of numbers differing by <= 15. Then you can use that to generate your dictionary easily.

def grouper(iterable):
    prev = None
    group = []
    for item in iterable:
        if prev is None or item - prev <= 15:
            group.append(item)
        else:
            yield group
            group = [item]
        prev = item
    if group:
        yield group

numbers = [123, 124, 128, 160, 167, 213, 215, 230, 245, 255, 257, 400, 401, 402, 430]
dict(enumerate(grouper(numbers), 1))

prints:

{1: [123, 124, 128],
 2: [160, 167],
 3: [213, 215, 230, 245, 255, 257],
 4: [400, 401, 402],
 5: [430]}

As a bonus, this lets you even group your runs for potentially-infinite lists (as long as they’re sorted, of course). You could also stick the index generation part into the generator itself (instead of using enumerate) as a minor enhancement.

More Related Contents:

  1. convert string to list [duplicate]
  2. How do i search for 2 of the same numbers in a list? (python) [closed]
  3. Unable to use convert strings into integers for a for loop
  4. How to remove an element from a list by index
  5. List of lists changes reflected across sublists unexpectedly
  6. Strange result when removing item from a list while iterating over it
  7. How to merge lists into a list of tuples?
  8. TypeError: ‘list’ object is not callable in python
  9. Difference between del, remove, and pop on lists
  10. Pythonic way to print list items
  11. Pythonic way to check if a list is sorted or not
  12. Detecting consecutive integers in a list [duplicate]
  13. List comprehension vs. lambda + filter
  14. Python – use list as function parameters
  15. Append to a list defined in a tuple – is it a bug? [duplicate]
  16. String to list in Python
  17. How to extract parameters from a list and pass them to a function call [duplicate]
  18. Sum of list of lists; returns sum list
  19. List of tuples to dictionary [duplicate]
  20. case-insensitive list sorting, without lowercasing the result?
  21. Group by multiple keys and summarize/average values of a list of dictionaries
  22. How to add element in Python to the end of list using list.insert?
  23. Append integer to beginning of list in Python [duplicate]
  24. Initializing 2D array in Python
  25. How to remove multiple indexes from a list at the same time? [duplicate]
  26. How to multiply all integers inside list [duplicate]
  27. Check if element exists in tuple of tuples
  28. Python: filtering lists by indices
  29. Filter a list of tuples based on condition
  30. How can I convert this string to list of lists? [duplicate]

Leave a Comment