![](/img/trans.png)
[英]Basic Python - finding difference between elements in a list - IndexError
[英]Finding Deltas - Difference between elements in a Python list
我有这段代码可以在列表中查找连续数字:
from itertools import groupby
from operator import itemgetter
a = [1,2,3,5,55,56]
def consc(b):
for k, g in groupby(enumerate(b), lambda (i,x):i-x):
print map(itemgetter(1), g)
consc(a)
输出:
[1, 2, 3]
[5]
[55, 56]
但是,我也希望能够寻找其他增量(1 到 10),例如 2 的差异将从同一列表中产生以下输出:
[1]
[2]
[3,5]
[55]
[56]
谢谢!
这实际上是一个非常简单的修改:
from itertools import groupby, count
from operator import itemgetter
a = [1,2,3,5,55,56]
def consc(b, step):
for k, g in groupby(zip(count(step=step), b), lambda (i, x): i-x):
print map(itemgetter(1), g)
consc(a, 2)
这使:
[1]
[2]
[3, 5]
[55]
[56]
我们不使用enumerate()
,而是使用带有所需值的步长的zip()
和count()
,这给出了想要的结果。
清理了一点:
from itertools import groupby, count
from operator import itemgetter
def _sub(item):
a, b = item
return a - b
def consecutive(iterable, step):
for _, g in groupby(zip(count(step=step), iterable), _sub):
yield map(itemgetter(1), g)
a = [1, 2, 3, 5, 55, 56]
print(list(consecutive(a, 2)))
在这里有一个生成器并使用更具描述性的名称是有意义的。 使用实际函数可以避免每次使用该函数时都重新声明它,就像lambda
。 通过避免使用已从语言中删除的参数解包,这也适用于 Python 3.x。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.