Python 将二维numpy阵列垂直拆分为不均匀子阵列

假设我有下面的

numpy

形状数组

（10，5）

，我想把它分成两个子数组：第一个包含前7行，第二个包含其余3行。如果我这样做：

x = np.arange(50).reshape(10, 5)
x1, y1 = np.vsplit(x, 2)

它将正好分成两半。如何将其设置为两个子阵列

（7,5）

和

（3,5）

使用

np.split（）

：

使用

np.split（）

：

---

我认为您应该使用花式索引，与切片不同，花式索引总是将数据复制到新数组中

n = 10; m = 5; i = 7
arr = np.arange(50).reshape(n, m)
arr7 = arr[np.ix_(range(i))]
arr3 = arr[np.ix_(range(i - n, 0, 1))]

---

我认为您应该使用花式索引，与切片不同，花式索引总是将数据复制到新数组中

n = 10; m = 5; i = 7
arr = np.arange(50).reshape(n, m)
arr7 = arr[np.ix_(range(i))]
arr3 = arr[np.ix_(range(i - n, 0, 1))]