获得列表元素之间差异的更好解决方案

时间:2014-07-22 14:34:38

标签: python

我没有寻找解决方案(我有两个;)),但是考虑到python的内部结构,比较每个解决方案的优缺点。谢谢!

对于同事,我们希望提取所有元素的两个连续列表元素之间的差异。所以,对于列表:

[1,2,4]

预期输出为:

[1,2]

(因为2-1 = 1,而4-2 = 2)。

我们提供了两种解决方案,我不确定它们是如何比较的。第一个是非常类似的,它将列表视为一个表,并减去两个连续列表元素之间的差异。

res = []
for i in range(0, len(a)-1):
    res.append(a[i+1] - a[i])

第二个是(对于列表" l"),我认为,更pythonic:

[j - i for i,j in zip(l[:-1], l[1:])]

尽管如此,构建列表的两个副本然后提取差异的效率却低得多? Python如何在内部处理它?<​​/ p>

感谢您的见解!

3 个答案:

答案 0 :(得分:3)

使用发电机:

def diff_elements(lst):
    """
    >>> list(diff_elements([]))
    []
    >>> list(diff_elements([1]))
    []
    >>> list(diff_elements([1, 2, 4, 7]))
    [1, 2, 3]
    """
    as_iter = iter(lst)
    last = next(as_iter)
    for value in as_iter:
        yield value - last
        last = value

这具有以下优良特性:

  1. 可读,
  2. 处理无限大的数据集。

答案 1 :(得分:0)

如果我理解你的问题,我建议你使用类似的东西:

diffList = lambda l: [(l[i] - l[i-1]) for i in range(1, len(l))]
answer = diffList( [ 1,2,4] )

此函数将为您提供一个列表,其中包含输入列表中所有连续元素之间的差异。

这个与你的第一种方法相似(并且仍然有些pythonic),它比第二种更有效。

答案 2 :(得分:0)

没有lambdas:

[l[i+1] - l[i] for i in range(len(l) - 1)]

例如:

>>> l = [1, 4, 8, 15, 16]
>>> [l[i+1] - l[i] for i in range(len(l) - 1)]
[3, 4, 7, 1]

快一点,你可以看到(编辑:在https://stackoverflow.com/a/2400875/1171280中添加投票最多的解决方案):

>>> import timeit
>>> 
>>> s = """\
...     l = [1, 4, 7, 15, 16]
...     [l[i+1] - l[i] for i in range(len(l) - 1)]
... """


>>> r = """\
...     l = [1, 4, 7, 15, 16]
...     [j - i for i,j in zip(l[:-1], l[1:])]
... """

>>> t = """\
...     l = [1, 4, 7, 15, 16]
...     [j-i for i, j in itertools.izip(l[:-1], l[1:])]
... """

>>> timeit.timeit(stmt=s, number=100000)
0.09615588188171387
>>> timeit.timeit(stmt=s, number=100000)
0.09774398803710938
>>> timeit.timeit(stmt=s, number=100000)
0.09683513641357422
#-------------
>>> timeit.timeit(stmt=r, number=100000)
0.14137601852416992
>>> timeit.timeit(stmt=r, number=100000)
0.12511301040649414
>>> timeit.timeit(stmt=r, number=100000)
0.12285017967224121
#-------------
>>> timeit.timeit(stmt=t, number=100000)
0.11506795883178711
>>> timeit.timeit(stmt=t, number=100000)
0.11677718162536621
>>> timeit.timeit(stmt=t, number=100000)
0.11829996109008789