Rolling or sliding window iterator?

后端 未结 23 1480
南方客
南方客 2020-11-21 05:23

I need a rolling window (aka sliding window) iterable over a sequence/iterator/generator. Default Python iteration can be considered a special case, where the window length

23条回答
  •  無奈伤痛
    2020-11-21 05:56

    #Importing the numpy library
    import numpy as np
    arr = np.arange(6) #Sequence
    window_size = 3
    np.lib.stride_tricks.as_strided(arr, shape= (len(arr) - window_size +1, window_size), 
    strides = arr.strides*2)
    
    """Example output:
    
      [0, 1, 2]
      [1, 2, 3]
      [2, 3, 4]
      [3, 4, 5]
    

    """

提交回复
热议问题