python - 具有可变增量的滑动窗口 - Python

Question

我正在尝试使用 Python 中的滑动窗口函数来比较一个很长的值列表。我为滑动窗口功能找到的代码如下：

from itertools import islice

idlist = [1, 2, 3, 4, 5, 6, 7, 8, 9]
list = []

def window(seq, n=2):
    "Returns a sliding window (of width n) over data from the iterable"
    "   s -> (s0,s1,...s[n-1]), (s1,s2,...,sn), ...                   "
    it = iter(seq)
    result = tuple(islice(it, n))
    if len(result) == n:
        yield result    
    for elem in it:
        result = result[1:] + (elem,)
        yield result

for i in window(idlist, n=2):
    list.append(i)

print list

我的问题是，我将如何修改此代码，以便可以将窗口的增量（生成每个元组后移动的量）从 1 更改为更大的整数，例如 5 或 50？我知道如何改变窗口的大小，但不知道增量。谢谢！

Answer 1

您不必更改增量，您可以获取每个第 n 个元素：

# taking every 3rd element moves the start by 3
print list(islice(window(idlist, n=2),None,None,3))

没有完全优化，但很简单。

Answer 2

提示：该next函数可用于从迭代器中获取下一个元素。您需要在每次迭代中获取并附加多个元素（我认为这就是困难；您肯定会看到如何将窗口的另一端向前移动一个不同的量:)）。

Answer 3

也许这可以解决问题

L=[1,2,3,4,5]

def window(L, n=2, jump=1):
    lenght = len(L)
    assert n <= lenght
    for i in range(0,lenght-n+1,jump):
        yield tuple(L[i:i+n])

A=[]
for i in window(L, n=3, jump=1):
    A.append(i)

print A