Indices multiples pour un tableau numpy: IndexError: impossible de forcer l'entrée de tranche de type numpy.ndarray en nombre entier

Question

Existe-t-il un moyen de faire plusieurs index dans un tableau numpy comme décrit ci-dessous?

arr=np.array([55, 2, 3, 4, 5, 6, 7, 8, 9]) 
arr[np.arange(0,2):np.arange(5,7)] 

output: 
IndexError: too many indices for array 

Desired output: 
array([55,2,3,4,5],[2,3,4,5,6]) 

Ce problème pourrait être semblable à calculer une moyenne mobile sur un tableau (mais je veux le faire sans aucune fonction qui est fournie).

Answer 1

Voilà une approche utilisant strides -

start_index = np.arange(0,2) 
L = 5  # Interval length 
n = arr.strides[0] 
strided = np.lib.stride_tricks.as_strided 
out = strided(arr[start_index[0]:],shape=(len(start_index),L),strides=(n,n)) 

run échantillon -

In [976]: arr 
Out[976]: array([55, 52, 13, 64, 25, 76, 47, 18, 69, 88]) 

In [977]: start_index 
Out[977]: array([2, 3, 4]) 

In [978]: L = 5 

In [979]: out 
Out[979]: 
array([[13, 64, 25, 76, 47], 
     [64, 25, 76, 47, 18], 
     [25, 76, 47, 18, 69]])