python列表添加数字_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文檔對此有一個(gè)非常簡潔的配方:
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]
如果要獲得完全相同的輸出,可以執(zhí)行以下操作:
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)]
編輯:該示例已在文檔中進(jìn)行了解釋,但也許我應(yīng)該對其進(jìn)行更多解釋:
解決方案的關(guān)鍵是在一定范圍內(nèi)進(jìn)行區(qū)分,以便連續(xù)的數(shù)字都出現(xiàn)在同一組中。
如果數(shù)據(jù)是:[2, 3, 4, 5, 12, 13, 14, 15, 16, 17] 然后groupby(enumerate(data), lambda (i,x):i-x)就相當(dāng)于以下內(nèi)容:
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
)
lambda函數(shù)從元素值中減去元素索引。因此,當(dāng)你在每個(gè)項(xiàng)目上應(yīng)用lambda時(shí)。你將獲得groupby的以下密鑰:
[-2, -2, -2, -2, -8, -8, -8, -8, -8, -8]
groupby按相等的鍵值將元素分組,因此前4個(gè)元素將被分組在一起,依此類推。
我希望這使它更具可讀性。
python 3 版本可能對初學(xué)者有所幫助
首先導(dǎo)入所需的庫
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]))
2020-02-14
總結(jié)
以上是生活随笔為你收集整理的python列表添加数字_Python-识别列表中的连续数字组的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 蒂法的魅力!《FF7RE》成系列Stea
- 下一篇: python输入直角三角形a、b、输出斜