如何分割一個numpy的陣列來獲得第一和最後兩行

Question

至於我可以看到它，它是不包括在該indexing, slicing and iterating SciPy的教程，所以讓我問在這裏：

說我已經

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

如何切片的順序排列得到第一和最後一行：

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

Answer 1

我不知道是否有做一個圓滑的方式。你可以列出的指標明確，當然：

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

或者使用r_幫忙，這很可能會更方便，如果我們希望從頭部或尾部更多行：

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

Answer 2

另外，您可以使用索引刪除爲

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