找到序列中长度为n的所有连续子序列

Question

我想在序列中找到长度为n的所有连续子序列.

例如,说n为3,序列为:

[0,1,7,3,4,5,10]

我想要一个可以产生输出的函数:

[[0,1,7],[1,7,3],[7,3,4],[3,4,5],[4,5,10]]

提前致谢!

Answer 1

>>> x = [0,1,7,3,4,5,10]
>>> n = 3
>>> zip(*(x[i:] for i in range(n)))
[(0, 1, 7), (1, 7, 3), (7, 3, 4), (3, 4, 5), (4, 5, 10)]

如果您希望结果是列表而不是元组列表,请使用map(list, zip(...)).

Answer 2

>>> x = [0,1,7,3,4,5,10]
>>> [x[n:n+3] for n in range(len(x)-2)]
[[0, 1, 7], [1, 7, 3], [7, 3, 4], [3, 4, 5], [4, 5, 10]]