3

I was thinking about this the other day. Say you have the list of numbers:

1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 19, 21, 22, 23, 999, 1000, 1001

is there any library or code snipped that will turn the following into something like:

1-13, 19, 21-23, 999-1001

In other words, reduce a full list of numbers to a bunch of ranges or so. I haven't been able to find anything. If no such thing exists, anyone have some ideas for an efficient implementation?

Cody Gray - on strike
  • 239,200
  • 50
  • 490
  • 574
cb22
  • 43
  • 1
  • 4
  • Related: [How to convert sequence of numbers in an array to range of numbers](https://stackoverflow.com/questions/2270910/how-to-convert-sequence-of-numbers-in-an-array-to-range-of-numbers) (but in JavaScript) – Cody Gray - on strike May 28 '20 at 23:48

3 Answers3

1
def get_groups(lst):
    slices = [i+1 for i, v in enumerate(zip(lst, l[1:])) if v[0] != v[1]-1] 
    slices = [0] + slices + [len(lst)]
    for start, end in zip(slices, slices[1:]):
        yield lst[start:end]

>>> list(get_groups([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 19, 21, 22, 23, 999, 1000, 1001]))
[[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13], [19], [21, 22, 23], [999, 1000, 1001]]

or

def get_ranges(lst):
    slices = [i+1 for i, v in enumerate(zip(lst, l[1:])) if v[0] != v[1]-1]
    slices = [0] + slices + [len(lst)]
    for start, end in zip(slices, slices[1:]):
        yield "%d-%d" % (lst[start], lst[end-1])

>>> list(get_ranges([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 19, 21, 22, 23, 999, 1000, 1001]))
['1-13', '19-19', '21-23', '999-1001']
Andrew Clark
  • 202,379
  • 35
  • 273
  • 306
  • 1
    +1 for generators and the fastest (worst case, at least): Steven - 0.954934120178 inspectorG4dget - 1.70556592941 Andrew - 0.000923871994019 That's on 1000 iterations of a random 1000 int list. – cb22 May 25 '11 at 20:48
0
def compress(nums):
    nums = [int(i) for i in nums.strip().split(',')]
    answer = []
    start = nums[0]
    prev = nums[0]
    for num in nums:
        if num-prev != 1:
            answer.append("%d-%d" %(start, prev))
            start = num
        prev = num
    answer.append("%d-%d" %(start, prev))
    return answer[1:]

>>> compress("1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 19, 21, 22, 23, 999, 1000, 1001")
['1-13', '19-19', '21-23', '999-1001']

Hope this helps

inspectorG4dget
  • 110,290
  • 27
  • 149
  • 241
0
l = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 19, 21, 22, 23, 999, 1000, 1001]
range_pos = []
start=0
end=0
for i in range(1,len(l)):
    if l[i] - l[i-1] == 1:
        end = i
    else:
        range_pos.append((start, end))
        start = end = i
range_pos.append((start, end))
ranges = ["%s-%s" % (l[s], l[e]) if s < e else str(l[s]) for (s, e) in range_pos]
print ', '.join(ranges)

Giving: 

1-13, 19, 21-23, 999-1001
Steven Rumbalski
  • 44,786
  • 9
  • 89
  • 119