將np.array分割成長度相同的子陣列

Question

我有一維不同長度的np.arrays。 什麼是最pythonic的方式將其分成長度相同的子數組（n = 384），並保留最後一個子數組（更短）？

Answer 1

假設x是一維數組，你可以這樣做：

x = x[:384 * (x.shape[0] // 384)].reshape(-1, 384) 

x現在有形狀(n, 384)。您可以使用x[i]訪問每個塊，其中0 <= i < n。

Answer 2

取決於「子陣列」的含義。

如果你想大小的列表384個ndarray對象，這不正是你要求：

np.split(x, np.arange(0, x.size, 384))[:-1] 

你可以用在nd.array()得到一個輸出相當於@COLDSPEED，但可能有點慢。

的真正的好處，這就是如果你想不同大小的一顆顆ndarrays連接成一個n x 384排列，列表的級聯可能比ndarray對象