如何将一个元素与排序列表中的下一个元素进行比较,并打印出它们的差异.任何帮助,将不胜感激.
Eg:
lst = [3.18,10.57,14.95,...]
10.57 - 3.18 = 7.39
14.95 - 10.57 = 4.38
...
Run Code Online (Sandbox Code Playgroud)
it = iter(lst)
it.next()
print [(x, y, y - x) for (x, y) in itertools.izip(lst, it)]
Run Code Online (Sandbox Code Playgroud)
如果您正在操作数值数据,请考虑使用numpy
import numpy as np
lst = [3.18,10.57,14.95]
arr = np.array(lst)
diff = np.diff(arr)
>>> diff
array([ 7.39, 4.38])
Run Code Online (Sandbox Code Playgroud)
如果必须,您可以将其转换回列表:
diff_list = list(diff)
Run Code Online (Sandbox Code Playgroud)
否则,您可以迭代它,就像迭代列表一样:
for item in diff:
print(item)
7.39
4.38
Run Code Online (Sandbox Code Playgroud)
编辑:我定时的五个解决方案彼此非常接近,所以选择一个更容易阅读的解决方案
t = timeit.Timer("[b - a for a, b in zip(l, l[1:])]", "l = range(int(1e6))")
print(t.timeit(1))
>>> 0.523894071579
t = timeit.Timer("list(np.diff(np.array(l)))", "import numpy as np; l = range(int(1e6))")
print(t.timeit(1))
>>> 0.484916915894
t = timeit.Timer("diffs = [l[x + 1] - l[x] for x in range(len(l) - 1)]", "l = range(int(1e6))")
print(t.timeit(1))
>>> 0.363043069839
t = timeit.Timer("[(x, y, y - x) for (x, y) in itertools.izip(l, it)]", "l = range(int(1e6)); it = iter(l); it.next()")
print(t.timeit(1))
>>> 0.54354596138
# pairwise solution
t = timeit.Timer("a, b = itertools.tee(l); next(b, None); [(x, y) for x, y in itertools.izip(a, b)]", "l = range(int(1e6));")
print(t.timeit(1))
>>> 0.477301120758
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
3302 次 |
| 最近记录: |