如何通过在最后一个元素处拆分数组，将1D数组拆分为NumPy中的2D数组？

Question

我有一个numpy数组，比如说

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

我想把它分成2d数组，这样最后一个元素就在它自己的数组中，就像这样

 ([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])]