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

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