Wie erstelle ich ein gleitendes Fenster mit einer 50% Überlappung mit einem numply Array?

Question

Sagen, ich habe ein Array wie folgt:

array([1, 2, 3, 4, 5, 5, 7, 8, 9, 10]) 

ich es

array([1, 2, 3, 4]), array([3, 4, 5, 6]), array([5, 6, 7, 8]), array([7, 8, 9, 10]) 

Hier wird die Fenstergröße 4 ändern wollen würde, und Schrittgröße würde 2 sein.

Wie kann ich dies in einer Python-Funktion tun?

Answer 1

Sie wollen numpy.lib.stride_tricks.as_strided

def window(a, w = 4, o = 2, copy = False): 
    sh = (a.size - w + 1, w) 
    st = a.strides * 2 
    view = np.lib.stride_tricks.as_strided(a, strides = st, shape = sh)[0::o] 
    if copy: 
     return view.copy() 
    else: 
     return view 

Out[]: 
array([[ 1, 2, 3, 4], 
     [ 3, 4, 5, 6], 
     [ 5, 6, 7, 8], 
     [ 7, 8, 9, 10]]) 

Seien Sie sicher, copy = True es zu setzen, wenn Sie an das fenster Array schreiben möchten, da sonst ist es ein Speicher-Sharing-Ansicht.