ระบุกลุ่มของตัวเลขต่อเนื่องในรายการ

ฉันต้องการระบุกลุ่มของตัวเลขต่อเนื่องในรายการเพื่อให้:

myfunc([2, 3, 4, 5, 12, 13, 14, 15, 16, 17, 20])

ผลตอบแทน:

[(2,5), (12,17), 20]

และสงสัยว่าวิธีที่ดีที่สุดในการทำเช่นนี้คืออะไร (โดยเฉพาะอย่างยิ่งหากมีบางสิ่งที่ฝังอยู่ใน Python)

แก้ไข: หมายเหตุเดิมทีฉันลืมที่จะพูดถึงว่าตัวเลขแต่ละตัวควรส่งคืนเป็นตัวเลขแต่ละตัวไม่ใช่ช่วง

more_itertools.consecutive_groups ถูกเพิ่มในเวอร์ชัน 4.0

การสาธิต

import more_itertools as mit


iterable = [2, 3, 4, 5, 12, 13, 14, 15, 16, 17, 20]
[list(group) for group in mit.consecutive_groups(iterable)]
# [[2, 3, 4, 5], [12, 13, 14, 15, 16, 17], [20]]

รหัส

เมื่อใช้เครื่องมือนี้เราสร้างฟังก์ชันตัวสร้างที่ค้นหาช่วงของตัวเลขที่ต่อเนื่องกัน

def find_ranges(iterable):
    """Yield range of consecutive numbers."""
    for group in mit.consecutive_groups(iterable):
        group = list(group)
        if len(group) == 1:
            yield group[0]
        else:
            yield group[0], group[-1]


iterable = [2, 3, 4, 5, 12, 13, 14, 15, 16, 17, 20]
list(find_ranges(iterable))
# [(2, 5), (12, 17), 20]

การใช้งานต้นฉบับเลียนแบบสูตรอาหารแบบคลาสสิก (ตามที่ @Nadia Alramli แสดงให้เห็น)

หมายเหตุ: more_itertoolsเป็นแพ็คเกจของ บริษัท อื่นที่สามารถติดตั้งได้ผ่านทางpip install more_itertools.

แก้ไข 2: เพื่อตอบสนองความต้องการใหม่ของ OP

ranges = []
for key, group in groupby(enumerate(data), lambda (index, item): index - item):
    group = map(itemgetter(1), group)
    if len(group) > 1:
        ranges.append(xrange(group[0], group[-1]))
    else:
        ranges.append(group[0])

เอาท์พุต:

[xrange(2, 5), xrange(12, 17), 20]

คุณสามารถแทนที่ xrange ด้วย range หรือคลาสที่กำหนดเองอื่น ๆ

เอกสาร Python มีสูตรที่เป็นระเบียบมากสำหรับสิ่งนี้:

from operator import itemgetter
from itertools import groupby
data = [2, 3, 4, 5, 12, 13, 14, 15, 16, 17]
for k, g in groupby(enumerate(data), lambda (i,x):i-x):
    print map(itemgetter(1), g)

เอาท์พุต:

[2, 3, 4, 5]
[12, 13, 14, 15, 16, 17]

หากคุณต้องการให้ได้ผลลัพธ์ที่เหมือนกันคุณสามารถทำได้:

ranges = []
for k, g in groupby(enumerate(data), lambda (i,x):i-x):
    group = map(itemgetter(1), g)
    ranges.append((group[0], group[-1]))

เอาต์พุต:

[(2, 5), (12, 17)]

แก้ไข:ตัวอย่างได้อธิบายไว้แล้วในเอกสาร แต่บางทีฉันควรอธิบายเพิ่มเติม:

กุญแจสำคัญในการแก้ปัญหาคือความแตกต่างของช่วงเพื่อให้ตัวเลขที่ติดต่อกันทั้งหมดปรากฏในกลุ่มเดียวกัน

หากข้อมูลเป็น: [2, 3, 4, 5, 12, 13, 14, 15, 16, 17] จากนั้นgroupby(enumerate(data), lambda (i,x):i-x)จะเทียบเท่ากับสิ่งต่อไปนี้:

groupby(
    [(0, 2), (1, 3), (2, 4), (3, 5), (4, 12),
    (5, 13), (6, 14), (7, 15), (8, 16), (9, 17)],
    lambda (i,x):i-x
)

ฟังก์ชันแลมบ์ดาจะลบดัชนีองค์ประกอบออกจากค่าองค์ประกอบ ดังนั้นเมื่อคุณใช้แลมด้าในแต่ละรายการ คุณจะได้รับคีย์ต่อไปนี้สำหรับ groupby:

[-2, -2, -2, -2, -8, -8, -8, -8, -8, -8]

groupby จัดกลุ่มองค์ประกอบตามค่าคีย์ที่เท่ากันดังนั้น 4 องค์ประกอบแรกจะถูกจัดกลุ่มเข้าด้วยกัน

ฉันหวังว่านี่จะทำให้อ่านง่ายขึ้น

python 3 เวอร์ชันอาจเป็นประโยชน์สำหรับผู้เริ่มต้น

นำเข้าไลบรารีที่ต้องการก่อน

from itertools import groupby
from operator import itemgetter

ranges =[]

for k,g in groupby(enumerate(data),lambda x:x[0]-x[1]):
    group = (map(itemgetter(1),g))
    group = list(map(int,group))
    ranges.append((group[0],group[-1]))

วิธีแก้ปัญหา "ไร้เดียงสา" ซึ่งฉันพบว่าอ่านได้ค่อนข้างน้อย

x = [2, 3, 4, 5, 12, 13, 14, 15, 16, 17, 22, 25, 26, 28, 51, 52, 57]

def group(L):
    first = last = L[0]
    for n in L[1:]:
        if n - 1 == last: # Part of the group, bump the end
            last = n
        else: # Not part of the group, yield current group and start a new
            yield first, last
            first = last = n
    yield first, last # Yield the last group


>>>print list(group(x))
[(2, 5), (12, 17), (22, 22), (25, 26), (28, 28), (51, 52), (57, 57)]

สมมติว่ารายการของคุณถูกจัดเรียง:

>>> from itertools import groupby
>>> def ranges(lst):
    pos = (j - i for i, j in enumerate(lst))
    t = 0
    for i, els in groupby(pos):
        l = len(list(els))
        el = lst[t]
        t += l
        yield range(el, el+l)


>>> lst = [2, 3, 4, 5, 12, 13, 14, 15, 16, 17]
>>> list(ranges(lst))
[range(2, 6), range(12, 18)]

นี่คือสิ่งที่ควรใช้งานได้โดยไม่ต้องนำเข้าใด ๆ :

def myfunc(lst):
    ret = []
    a = b = lst[0]                           # a and b are range's bounds

    for el in lst[1:]:
        if el == b+1: 
            b = el                           # range grows
        else:                                # range ended
            ret.append(a if a==b else (a,b)) # is a single or a range?
            a = b = el                       # let's start again with a single
    ret.append(a if a==b else (a,b))         # corner case for last single/range
    return ret

โปรดทราบว่ารหัสที่ใช้groupbyไม่ทำงานตามที่ระบุใน Python 3 ดังนั้นให้ใช้สิ่งนี้

for k, g in groupby(enumerate(data), lambda x:x[0]-x[1]):
    group = list(map(itemgetter(1), g))
    ranges.append((group[0], group[-1]))

สิ่งนี้ไม่ได้ใช้ฟังก์ชั่นมาตรฐาน - มันเป็นเพียงแค่การเพิ่มขึ้นของอินพุต แต่ควรใช้งานได้:

def myfunc(l):
    r = []
    p = q = None
    for x in l + [-1]:
        if x - 1 == q:
            q += 1
        else:
            if p:
               if q > p:
                   r.append('%s-%s' % (p, q))
               else:
                   r.append(str(p))
            p = q = x
    return '(%s)' % ', '.join(r)

โปรดทราบว่าอินพุตต้องมีเฉพาะตัวเลขบวกจากน้อยไปหามาก คุณควรตรวจสอบความถูกต้องของข้อมูล แต่รหัสนี้จะถูกละไว้เพื่อความชัดเจน

การใช้groupbyและcountจากitertoolsช่วยให้เรามีทางออกสั้น ๆ แนวคิดก็คือในลำดับที่เพิ่มขึ้นความแตกต่างระหว่างดัชนีและค่าจะยังคงเหมือนเดิม

ในการติดตามดัชนีเราสามารถใช้itertools.countซึ่งทำให้โค้ดสะอาดขึ้นโดยใช้enumerate:

from itertools import groupby, count

def intervals(data):
    out = []
    counter = count()

    for key, group in groupby(data, key = lambda x: x-next(counter)):
        block = list(group)
        out.append([block[0], block[-1]])
    return out

ผลลัพธ์ตัวอย่างบางส่วน:

print(intervals([0, 1, 3, 4, 6]))
# [[0, 1], [3, 4], [6, 6]]

print(intervals([2, 3, 4, 5]))
# [[2, 5]]

นี่คือคำตอบที่ฉันคิดขึ้น ฉันกำลังเขียนโค้ดเพื่อให้คนอื่นเข้าใจดังนั้นฉันจึงค่อนข้างใช้ชื่อตัวแปรและความคิดเห็น

ก่อนอื่นฟังก์ชั่นตัวช่วยด่วน:

def getpreviousitem(mylist,myitem):
    '''Given a list and an item, return previous item in list'''
    for position, item in enumerate(mylist):
        if item == myitem:
            # First item has no previous item
            if position == 0:
                return None
            # Return previous item    
            return mylist[position-1] 

จากนั้นรหัสจริง:

def getranges(cpulist):
    '''Given a sorted list of numbers, return a list of ranges'''
    rangelist = []
    inrange = False
    for item in cpulist:
        previousitem = getpreviousitem(cpulist,item)
        if previousitem == item - 1:
            # We're in a range
            if inrange == True:
                # It's an existing range - change the end to the current item
                newrange[1] = item
            else:    
                # We've found a new range.
                newrange = [item-1,item]
            # Update to show we are now in a range    
            inrange = True    
        else:   
            # We were in a range but now it just ended
            if inrange == True:
                # Save the old range
                rangelist.append(newrange)
            # Update to show we're no longer in a range    
            inrange = False 
    # Add the final range found to our list
    if inrange == True:
        rangelist.append(newrange)
    return rangelist

ตัวอย่างการรัน:

getranges([2, 3, 4, 5, 12, 13, 14, 15, 16, 17])

ผลตอบแทน:

[[2, 5], [12, 17]]

import numpy as np

myarray = [2, 3, 4, 5, 12, 13, 14, 15, 16, 17, 20]
sequences = np.split(myarray, np.array(np.where(np.diff(myarray) > 1)[0]) + 1)
l = []
for s in sequences:
    if len(s) > 1:
        l.append((np.min(s), np.max(s)))
    else:
        l.append(s[0])
print(l)

เอาท์พุต:

[(2, 5), (12, 17), 20]

การใช้รายการความเข้าใจ numpy +:
ด้วยฟังก์ชัน numpy diff รายการเวกเตอร์อินพุตที่ตามมาซึ่งความแตกต่างไม่เท่ากับหนึ่งสามารถระบุได้ ต้องพิจารณาจุดเริ่มต้นและจุดสิ้นสุดของเวกเตอร์อินพุต

import numpy as np
data = np.array([2, 3, 4, 5, 12, 13, 14, 15, 16, 17, 20])

d = [i for i, df in enumerate(np.diff(data)) if df!= 1] 
d = np.hstack([-1, d, len(data)-1])  # add first and last elements 
d = np.vstack([d[:-1]+1, d[1:]]).T

print(data[d])

เอาท์พุต:

 [[ 2  5]   
  [12 17]   
  [20 20]]

หมายเหตุ: คำขอที่ว่าแต่ละหมายเลขควรได้รับการปฏิบัติที่แตกต่างกัน (ส่งคืนเป็นแต่ละหมายเลขไม่ใช่ช่วง) ถูกละไว้ สามารถเข้าถึงได้โดยการประมวลผลผลลัพธ์เพิ่มเติม โดยปกติสิ่งนี้จะทำให้สิ่งต่างๆซับซ้อนขึ้นโดยไม่ได้รับประโยชน์ใด ๆ

โซลูชันสั้น ๆ ที่ใช้งานได้โดยไม่ต้องนำเข้าเพิ่มเติม ยอมรับการทำซ้ำเรียงลำดับอินพุตที่ไม่ได้เรียงลำดับและลบรายการที่ซ้ำกัน:

def ranges(nums):
    nums = sorted(set(nums))
    gaps = [[s, e] for s, e in zip(nums, nums[1:]) if s+1 < e]
    edges = iter(nums[:1] + sum(gaps, []) + nums[-1:])
    return list(zip(edges, edges))

ตัวอย่าง:

>>> ranges([2, 3, 4, 7, 8, 9, 15])
[(2, 4), (7, 9), (15, 15)]

>>> ranges([-1, 0, 1, 2, 3, 12, 13, 15, 100])
[(-1, 3), (12, 13), (15, 15), (100, 100)]

>>> ranges(range(100))
[(0, 99)]

>>> ranges([0])
[(0, 0)]

>>> ranges([])
[]

นี่เหมือนกับโซลูชันของ @ dansalmo ซึ่งฉันพบว่าน่าทึ่งแม้ว่าจะอ่านและนำไปใช้ยากสักหน่อย (เนื่องจากไม่ได้กำหนดให้เป็นฟังก์ชัน)

โปรดทราบว่าสามารถแก้ไขได้อย่างง่ายดายเพื่อคายช่วงเปิด "ดั้งเดิม" ออก[start, end)โดยเช่นการแก้ไขคำสั่ง return:

    return [(s, e+1) for s, e in zip(edges, edges)]

ฉันคัดลอกคำตอบนี้มาจากคำถามอื่นที่ทำเครื่องหมายว่าซ้ำกับคำถามนี้โดยมีจุดประสงค์เพื่อให้ค้นหาได้ง่ายขึ้น (หลังจากที่ฉันค้นหาหัวข้อนี้อีกครั้งในตอนแรกพบเฉพาะคำถามที่นี่ในตอนแรกและไม่พอใจกับคำตอบ ให้).

เวอร์ชันของMark Byers , Andrea Ambu , SilentGhost , Nadia Alramliและtruppoนั้นง่ายและรวดเร็ว เวอร์ชัน 'truppo' สนับสนุนให้ฉันเขียนเวอร์ชันที่ยังคงลักษณะการทำงานที่ว่องไวเหมือนเดิมในขณะที่จัดการขนาดขั้นตอนอื่นที่ไม่ใช่ 1 (และแสดงรายการเป็นองค์ประกอบเดี่ยวที่ไม่ขยายมากกว่า 1 ขั้นตอนตามขนาดขั้นตอนที่กำหนด) มันจะได้รับที่นี่

>>> list(ranges([1,2,3,4,3,2,1,3,5,7,11,1,2,3]))
[(1, 4, 1), (3, 1, -1), (3, 7, 2), 11, (1, 3, 1)]