разбирать строку целых множеств с интервалами в списке - PullRequest
6 голосов
/ 18 апреля 2011

У меня "2,5,7-9,12" строка.

Я хочу получить список [2, 5, 7, 8, 9, 12] из него.

Есть ли в Python встроенная функция для него?

Спасибо.

UPD. Полагаю, прямой ответ Нет . В любом случае, спасибо за ваши "отрывки". Используя один, предложенный Свен Марнах .

Ответы [ 6 ]

15 голосов
/ 18 апреля 2011
s = "2,5,7-9,12"
ranges = (x.split("-") for x in s.split(","))
print [i for r in ranges for i in range(int(r[0]), int(r[-1]) + 1)]

печать

[2, 5, 7, 8, 9, 12]
5 голосов
/ 18 апреля 2011
s = "2,5,7-9,12"
result = list()

for item in s.split(','):
    if '-' in item:
        x,y = item.split('-')
        result.extend(range(int(x), int(y)+1))
    else:
        result.append(int(item))

print result
2 голосов
/ 20 октября 2015

Эта версия обрабатывает произвольные пробелы, перекрывающиеся диапазоны, диапазоны вне порядка и целые отрицательные числа:

from itertools import chain

def group_to_range(group):
  group = ''.join(group.split())
  sign, g = ('-', group[1:]) if group.startswith('-') else ('', group)
  r = g.split('-', 1)
  r[0] = sign + r[0]
  r = sorted(int(__) for __ in r)
  return range(r[0], 1 + r[-1])

def rangeexpand(txt):
  ranges = chain.from_iterable(group_to_range(__) for __ in txt.split(','))
  return sorted(set(ranges))


>>> rangeexpand('-6,-3--1,3-5,7-11,14,15,17-20')
[-6, -3, -2, -1, 3, 4, 5, 7, 8, 9, 10, 11, 14, 15, 17, 18, 19, 20]
>>> rangeexpand('1-4,6,3-2, 11, 8 - 12,5,14-14')
[1, 2, 3, 4, 5, 6, 8, 9, 10, 11, 12, 14]
2 голосов
/ 18 апреля 2011

Я бы определил функцию:

def make_range(s):
    out = []
    s = s.split(',')
    for n in s:
        if '-' in n:
            n = n.split('-')
            for i in range(int(n[0]), int(n[1]) + 1):
                out.append(i)
        else:
            out.append(int(n))
    return out

print make_range("2,5,7-9,12")
#output [2, 5, 7, 8, 9, 12]
1 голос
/ 18 апреля 2011

Я не знаю ни одной встроенной функции, которая бы это делала.Следующее не особенно элегантно, но выполняет свою работу:

s = "2,5,7-9,12"
ret = []
for tok in s.split(","):
  val = map(int, tok.split("-"))
  if len(val) == 1:
    ret += val
  else:
    ret += range(val[0], val[1] + 1)
print ret

Одной из областей, где это решение может потребоваться, является обработка отрицательных чисел (из вашего вопроса не совсем ясно, могут ли отрицательные числапоявляются на входе).

1 голос
/ 18 апреля 2011

Не то, чтобы я знал, но вы можете легко сделать свой собственный:

  1. Создать список результатов.
  2. Разделить строки на , и начать итерацию по результату.
    1. Если текущая строка содержит -, добавьте диапазон в список.
    2. Если текущая строка является числом, добавьте ее в список.
    3. В противном случае возвращает ошибку.
  3. Вернуть список.
...