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] 

基本上，现在很容易对数据进行分片。如果需要，可以使用类似的逻辑轻松更改切点。