Expand a range which looks like: "1-3,6,8-10" to [1,2,3, 6, 8,9,10]

Question

I am trying to add an option to my program which allow the user to choose which steps of the program he wants to do.

I would like to be able to parse a string like "1-3,6,8-10" and get [1, 2, 3, 6, 8, 9, 10].

Do you know if something in Python which is doing that already exists ?

Answer 1

This function does what you asked. It assumes no negative numbers are used, otherwise it needs some changes to support that case.

def mixrange(s):
    r = []
    for i in s.split(','):
        if '-' not in i:
            r.append(int(i))
        else:
            l,h = map(int, i.split('-'))
            r+= range(l,h+1)
    return r


print mixrange('1-3,6,8-10')

Answer 2

One way using list comprehensions:

s = "1-3,6,8-10"
x = [ss.split('-') for ss in s.split(',')]
x = [range(int(i[0]),int(i[1])+1) if len(i) == 2 else i for i in x]
print([int(item) for sublist in x for item in sublist])

Outputs:

[1, 2, 3, 6, 8, 9, 10]