Come posso tagliare un array di NumPy per ottenere sia le prime e le ultime due file

Question

Per quanto posso vedere, non è coperto nel indexing, slicing and iterating esercitazione SciPy, quindi permettetemi di chiedere qui:

Say ho

x = np.array([[1,2],[3,4],[5,6],[7,8],[9,0]]) 
x: array([[1, 2], 
      [3, 4], 
      [5, 6], 
      [7, 8], 
      [9, 0]]) 

Come faccio la fetta di array in ordine ottenere sia la prima e l'ultima fila:

y: array([[1, 2], 
      [3, 4],    
      [7, 8], 
      [9, 0]]) 

Answer 1

non so se c'è un modo per farlo chiazza di petrolio. Si potrebbe elencare gli indici in modo esplicito, ovviamente:

>>> x[[0,1,-2,-1]] 
array([[1, 2], 
     [3, 4], 
     [7, 8], 
     [9, 0]]) 

Oppure utilizzare r_ per aiutare, che probabilmente sarebbe più conveniente se volevamo più righe dal capo o la coda:

>>> x[np.r_[0:2, -2:0]] 
array([[1, 2], 
     [3, 4], 
     [7, 8], 
     [9, 0]]) 

Answer 2

In alternativa, è possibile utilizzare gli indici per rimuovere come

mask = np.ones(len(x), dtype=np.bool) 
mask[2:3] = False # that you want to remove 
y = x[mask]