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

实现这一目标的最简单方法是什么？现在我正在复制原始数组两次并删除我不需要的部分。似乎应该有一种简单的方法将阵列分成不等大小的部分。 split和vsplit似乎只能分成相同大小的块。任何帮助表示赞赏。

Answer 1

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

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