将numpy数组拆分为两个不同大小的子集

Question

我有一个形状为(400, 3, 3, 3)的numpy数组，我想把它分成两部分，这样我就得到了(100, 3, 3, 3)和{}这样的数组。在

我在玩numpysplit方法，例如：

subsets = np.array_split(arr, 2)

这给了我想要的，但是它将原始数组分成两部分，大小相同，我不知道如何指定这些大小。用一些索引可能很容易（我想），但我不知道怎么做。在

Answer 1

正如我在评论中提到的，您可以使用Ellipsis符号来指定所有轴：

x, y = arr[:100, ...], arr[100:, ...]