我有一个形状(400, 3, 3, 3)的numpy数组，我想把它分成两部分，所以我会得到像(100, 3, 3, 3)和(300, 3, 3, 3)这样的数组。
我正在使用numpy split方法，例如：
subsets = np.array_split(arr, 2)
它给了我想要的东西，但是它将原始数组分成相同大小的两半，我不知道如何指定这些大小。一些索引（我猜）可能很容易，但我不知道该怎么做。
答案 0 :(得分：3)
如我的评论所述，您可以使用Ellipsis表示法指定所有轴：
x, y = arr[:100, ...], arr[100:, ...]