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

Question

我有一个形状的 numpy 数组，(400, 3, 3, 3)我想将它分成两部分，所以我会得到像(100, 3, 3, 3)和 的数组(300, 3, 3, 3)。

我在玩 numpysplit方法，例如：

subsets = np.array_split(arr, 2)

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

Answer 1

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

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