如何通过在最后一个元素拆分阵列拆分一维数组到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])]