Find all consecutive sub-sequences of length n in a sequence

前端 未结 4 1296
野的像风
野的像风 2020-12-10 18:03

I want to find all consecutive sub-sequences of length n in a sequence.

E.g. say n was 3 and the sequence was:

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

I w

4条回答
  •  长情又很酷
    2020-12-10 19:02

    >>> 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)]
    

    If you want the result to be a list of lists instead of list of tuples, use map(list, zip(...)).

提交回复
热议问题