リスト内の連続番号のグループを識別したいので、次のようにします。
myfunc([2, 3, 4, 5, 12, 13, 14, 15, 16, 17, 20])
戻り値:
[(2,5), (12,17), 20]
そして、これを行うための最良の方法は何であるか疑問に思っていました(特にPythonに何かが組み込まれている場合)。
編集:注私は当初、個々の数値は範囲ではなく個別の数値として返される必要があることを言及するのを忘れていました。
リスト内の連続番号のグループを識別したいので、次のようにします。
myfunc([2, 3, 4, 5, 12, 13, 14, 15, 16, 17, 20])
戻り値:
[(2,5), (12,17), 20]
そして、これを行うための最良の方法は何であるか疑問に思っていました(特にPythonに何かが組み込まれている場合)。
編集:注私は当初、個々の数値は範囲ではなく個別の数値として返される必要があることを言及するのを忘れていました。
編集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]
Thengroupby(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]))
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
。
私が少なくともいくらか読みやすいと思う「素朴な」解決策。
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
を使用するコードはPython3で指定されているように機能しないことに注意してください。したがって、これを使用してください。
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)
入力には昇順の正の数のみが含まれている必要があることに注意してください。入力を検証する必要がありますが、わかりやすくするためにこのコードは省略されています。
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]
groupby
and count
fromを使用すると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]]
この方法は、私がここで見たどの回答よりも簡単だと思います(編集:Pleastryからのコメントに基づいて修正):
data = [2, 3, 4, 5, 12, 13, 14, 15, 16, 17, 20]
starts = [x for x in data if x-1 not in data and x+1 in data]
ends = [x for x in data if x-1 in data and x+1 not in data and x not in starts]
singles = [x for x in data if x-1 not in data and x+1 not in data]
list(zip(starts, ends)) + singles
出力:
[(2, 5), (12, 17), 20]
これが私が思いついた答えです。私は他の人が理解できるようにコードを書いているので、変数名とコメントをかなり冗長にしています。
まず、クイックヘルパー関数:
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]]
numpy +理解リストの使用:
numpy diff関数を使用すると、結果として生じる差が1に等しくない入力ベクトルエントリを識別できます。入力ベクトルの開始と終了を考慮する必要があります。
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]]
注:個々の数値を異なる方法で処理する必要があるという要求(範囲ではなく個別として返される)は省略されました。これは、結果をさらに後処理することで達成できます。通常、これは何の利益も得ずに物事をより複雑にします。
これは、読みやすさを優先しようとした私の方法です。グループに値が1つしかない場合は、同じ値のタプルが返されることに注意してください。これは、投稿する2番目のスニペットで簡単に修正できます。
def group(values):
"""return the first and last value of each continuous set in a list of sorted values"""
values = sorted(values)
first = last = values[0]
for index in values[1:]:
if index - last > 1: # triggered if in a new group
yield first, last
first = index # update first only if in a new group
last = index # update last on every iteration
yield first, last # this is needed to yield the last set of numbers
テストの結果は次のとおりです。
values = [0, 5, 6, 7, 12, 13, 21, 22, 23, 24, 25, 26, 30, 44, 45, 50]
result = list(group(values))
print(result)
結果=[(0, 0), (5, 7), (12, 13), (21, 26), (30, 30), (44, 45), (50, 50)]
グループ内の単一の値の場合に単一の値のみを返したい場合は、結果に条件付きチェックを追加するだけです。
def group(values):
"""return the first and last value of each continuous set in a list of sorted values"""
values = sorted(values)
first = last = values[0]
for index in values[1:]:
if index - last > 1: # triggered if in a new group
if first == last:
yield first
else:
yield first, last
first = index # update first only if in a new group
last = index # update last on every iteration
if first == last:
yield first
else:
yield first, last
結果=[0, (5, 7), (12, 13), (21, 26), 30, (44, 45), 50]
追加のインポートなしで機能する短いソリューション。反復可能なものを受け入れ、ソートされていない入力をソートし、重複するアイテムを削除します。
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)]
最善のアプローチではありませんが、これが私の2セントです
def getConsecutiveValues2(arr):
x = ""
final = []
end = 0
start = 0
for i in range(1,len(arr)) :
if arr[i] - arr[i-1] == 1 :
end = i
else :
print(start,end)
final.append(arr[start:end+1])
start = i
if i == len(arr) - 1 :
final.append(arr[start:end+1])
return final
x = [1,2,3,5,6,8,9,10,11,12]
print(getConsecutiveValues2(x))
>> [[1, 2, 3], [5, 6], [8, 9, 10, 11]]
この実装は、定期的または不定期の手順で機能します
私は同じことを達成する必要がありましたが、ステップが不規則になる可能性があるわずかな違いがあります。これは私の実装です
def ranges(l):
if not len(l):
return range(0,0)
elif len(l)==1:
return range(l[0],l[0]+1)
# get steps
sl = sorted(l)
steps = [i-j for i,j in zip(sl[1:],sl[:-1])]
# get unique steps indexes range
groups = [[0,0,steps[0]],]
for i,s in enumerate(steps):
if s==groups[-1][-1]:
groups[-1][1] = i+1
else:
groups.append( [i+1,i+1,s] )
g2 = groups[-2]
if g2[0]==g2[1]:
if sl[i+1]-sl[i]==s:
_=groups.pop(-2)
groups[-1][0] = i
# create list of ranges
return [range(sl[i],sl[j]+s,s) if s!=0 else [sl[i]]*(j+1-i) for i,j,s in groups]
これが例です
from timeit import timeit
# for regular ranges
l = list(range(1000000))
ranges(l)
>>> [range(0, 1000000)]
l = list(range(10)) + list(range(20,25)) + [1,2,3]
ranges(l)
>>> [range(0, 2), range(1, 3), range(2, 4), range(3, 10), range(20, 25)]
sorted(l);[list(i) for i in ranges(l)]
>>> [0, 1, 1, 2, 2, 3, 3, 4, 5, 6, 7, 8, 9, 20, 21, 22, 23, 24]
>>> [[0, 1], [1, 2], [2, 3], [3, 4, 5, 6, 7, 8, 9], [20, 21, 22, 23, 24]]
# for irregular steps list
l = [1, 3, 5, 7, 10, 11, 12, 100, 200, 300, 400, 60, 99, 4000,4001]
ranges(l)
>>> [range(1, 9, 2), range(10, 13), range(60, 138, 39), range(100, 500, 100), range(4000, 4002)]
## Speed test
timeit("ranges(l)","from __main__ import ranges,l", number=1000)/1000
>>> 9.303160999934334e-06