如何通過在最後一個元素拆分陣列拆分一維數組到NumPy的二維數組？

Question

我有一個numpy的陣列，說

([1,2,3,4,5,6,7]) 

我想將它分割成一個二維數組，使得最後一個元素是在它自己的陣列，像這樣

([1,2,3,4,5,6],[7]) 

我究竟會怎麼做這個？

Answer 1

使用np.split -

np.split(a,[-1]) 

採樣運行 -

In [105]: a 
Out[105]: array([1, 2, 3, 4, 5, 6, 7]) 

In [106]: np.split(a,[-1]) 
Out[106]: [array([1, 2, 3, 4, 5, 6]), array([7])]