Python - 將numpy數組分割成不同大小的部分

Question

我試圖打破沿着某一行的numpy數組，以便最終得到「頂部」和「底部」。

例如

[[2 1 3 2] 
[1 6 7 2] 
[2 8 6 3] 
[3 4 2 2]] 

top = [2 1 3 2] 

bottom = [[1 6 7 2] 
      [2 8 6 3] 
      [3 4 2 2]] 

什麼是完成這一任務的最簡單的方法？現在我正在複製原始數組兩次並刪除不需要的部分。似乎應該有一種簡單的方法將數組分成不同大小的部分。拆分和vsplit似乎只能拆分成相同大小的塊。任何幫助表示讚賞。

Answer 1

top=data[0] 
bottom=data[1:3] 

基本上，現在很容易對數據進行分片。如果需要，可以使用類似的邏輯輕鬆更改切點。