2011-01-07 21 views
18

Czy istnieje coś istniejącego w Pythonie, które można przekształcić rosnącą listę liczb całkowitych w liście zakresukonwersji listę liczb całkowitych w zakresie w Pythonie

Np biorąc pod uwagę zbiór {0, 1, 2, 3, 4, 7, 8, 9, 11} Chcę uzyskać {{0,4}, {7,9}, {11,11}}.

mogę napisać program, aby to zrobić, ale chcę wiedzieć, czy jest wbudowana funkcja w Pythonie

+1

Prawie to samo pytanie zostało zadane i odpowiedzi http://stackoverflow.com/questions/3429510/pythonic-way-to-convert-a-list-of-integers-into-a-string-of -comma-separ-range/3430231 # 3430231 – Apalala

+1

'>>> import this' – Apalala

+0

Cóż, mogę powiedzieć z przekonaniem, że nie znam takiej funkcji. O wiele trudniej jest powiedzieć z przekonaniem, że coś, czego nie znam, nie istnieje ... –

Odpowiedz

25

Korzystanie itertools.groupby produkuje zwięzły ale trudne realizacji:

import itertools 

def ranges(i): 
    for a, b in itertools.groupby(enumerate(i), lambda (x, y): y - x): 
     b = list(b) 
     yield b[0][1], b[-1][1] 

print list(ranges([0, 1, 2, 3, 4, 7, 8, 9, 11])) 

wyjściowa:

[(0, 4), (7, 9), (11, 11)] 
+1

To jest naprawdę przydatne, zastanawiam się, czy mógłbyś wyjaśnić, jak działa ta metoda, abym mógł zrozumieć funkcjonalność. byłoby to wspaniale, jeśli to możliwe. – openCivilisation

+0

Aby obsłużyć nieunikalne i niesegregowane wejście "i" wejściowe z "posortowane (zestaw (i))", zobacz: https://stackoverflow.com/a/43091576/1201614 – luca

+0

Ten przepis jest również dostępny w 'more_itertools .consequive_groups'. Zobacz demonstrację [tutaj] (https://stackoverflow.com/a/47642650/4531270). – pylang

1

Nic wbudowany, lub w jakichkolwiek bibliotek, które znam. Niezbyt pomocna, wiem, ale nigdy nie spotkałem się z niczym, co się chce.

Oto kilka pomysłów na swój program Atleast (w C++, ale może dać ci kilka innych pomysłów):

Converting sets of integers into ranges

1

W przypadku nie ma takiej funkcji w Pythonie, tutaj jest implementacją

p = [] 
last = -2                
start = -1 

for item in list: 
    if item != last+1:       
     if start != -1: 
      p.append([start, last]) 
     start = item 
    last = item 

p.append([start, last]) 
2

generator:

def ranges(p): 
    q = sorted(p) 
    i = 0 
    for j in xrange(1,len(q)): 
     if q[j] > 1+q[j-1]: 
      yield (q[i],q[j-1]) 
      i = j 
    yield (q[i], q[-1]) 

sample = [0, 1, 2, 3, 4, 7, 8, 9, 11] 
print list(ranges(sample)) 
print list(ranges(reversed(sample))) 
print list(ranges([1])) 
print list(ranges([2,3,4])) 
print list(ranges([0,2,3,4])) 
print list(ranges(5*[1])) 

P roduces te wyniki:

[(0, 4), (7, 9), (11, 11)] 
[(0, 4), (7, 9), (11, 11)] 
[(1, 1)] 
[(2, 4)] 
[(0, 0), (2, 4)] 
[(1, 1)] 

Zauważ, że działa z powtarzającymi się numerami sprężonego. Nie wiem, czy tego chcesz. Jeśli nie, zmień > na !=.

Rozumiem twoje pytanie. Zaglądałem do itertools i próbowałem wymyślić rozwiązanie, które można by zrobić w kilku liniach Pythona, który zakwalifikowałby jako "prawie wbudowany", ale nie mogłem wymyślić niczego.

7

Można użyć list comprehension z generator expression i kombinacji enumerate() i itertools.groupby():

>>> import itertools 
>>> l = [0, 1, 2, 3, 4, 7, 8, 9, 11] 
>>> [[t[0][1], t[-1][1]] for t in 
... (tuple(g[1]) for g in itertools.groupby(enumerate(l), lambda (i, x): i - x))] 
[[0, 4], [7, 9], [11, 11]] 

pierwsze enumerate() zbudują krotki z elementami list a ich odpowiednie współczynniki:

>>> [t for t in enumerate(l)] 
[(0, 0), (1, 1), (2, 2), (3, 3), (4, 4), (5, 7), (6, 8), (7, 9), (8, 11)] 

Następnie groupby() będzie grupy te krotki, wykorzystując różnicę między indeksie i wartości (która będzie taki sam dla kolejnych wartości)

>>> [tuple(g[1]) for g in itertools.groupby(enumerate(l), lambda (i, x): i - x)] 
[((0, 0), (1, 1), (2, 2), (3, 3), (4, 4)), ((5, 7), (6, 8), (7, 9)), ((8, 11),)] 

Stamtąd, tylko trzeba zbudować list od wartości pierwszej i ostatniej krotki z każdej grupy (który będzie taki sam, jeśli grupa zawiera tylko jedną pozycję).

Można również użyć [(t[0][1], t[-1][1]) ...] zbudować listę krotek Zakres zamiast zagnieżdżonych list, a nawet ((t[0][1], t[-1][1]) ...) obrócić całe wyrażenie w iterowalny generator która leniwie wybudować krotki gama na bieżąco.

1

put krótsze:

ranges=lambda l:map(lambda x:(x[0][1],x[-1][1]),map(lambda (x,y):list(y),itertools.groupby(enumerate(l),lambda (x,y):x-y))) 
+2

Shorter nie jest poprawą, moim zdaniem. – madth3

1

Generowanie pary zakres:

def ranges(lst): 
    s = e = None 
    r = [] 
    for i in sorted(lst): 
     if s is None: 
      s = e = i 
     elif i == e or i == e + 1: 
      e = i 
     else: 
      r.append((s, e)) 
      s = e = i 
    if s is not None: 
     r.append((s, e)) 
    return r 

przykład:

>>> lst = [1, 5, 6, 7, 12, 15, 16, 17, 18, 30] 
>>> print repr(ranges(lst)) 
[(1, 1), (5, 7), (12, 12), (15, 18), (30, 30)] 

jako generator:

def gen_ranges(lst): 
    s = e = None 
    for i in sorted(lst): 
     if s is None: 
      s = e = i 
     elif i == e or i == e + 1: 
      e = i 
     else: 
      yield (s, e) 
      s = e = i 
    if s is not None: 
     yield (s, e) 

przykład:

>>> lst = [1, 5, 6, 7, 12, 15, 16, 17, 18, 30] 
>>> print repr(','.join(['%d' % s if s == e else '%d-%d' % (s, e) for (s, e) in gen_ranges(lst)])) 
'1,5-7,12,15-18,30' 
3

Jest to postęp w stosunku do bardzo elegancki @juanchopanza answer. Ten obejmuje nieunikatowa i non-sortowane wejście i jest python3 kompatybilny też:

import itertools 

def to_ranges(iterable): 
    iterable = sorted(set(iterable)) 
    for key, group in itertools.groupby(enumerate(iterable), 
             lambda t: t[1] - t[0]): 
     group = list(group) 
     yield group[0][1], group[-1][1] 

Przykład:

>>> x 
[44, 45, 2, 56, 23, 11, 3, 4, 7, 9, 1, 2, 2, 11, 12, 13, 45] 

>>> print(list(to_ranges(x))) 
[(1, 4), (7, 7), (9, 9), (11, 13), (23, 23), (44, 45), (56, 56)] 
0

myślę pozostałe odpowiedzi są trudne do zrozumienia, a prawdopodobnie nieskuteczne. Mam nadzieję, że jest to łatwiejsze i szybsze.

def ranges(ints): 
    ints = sorted(set(ints)) 
    range_start = previous_number = ints[0] 
    for number in ints[1:]: 
     if number == previous_number + 1: 
      previous_number = number 
     else: 
      yield range_start, previous_number 
      range_start = previous_number = number 
    yield range_start, previous_number 
Powiązane problemy