python - 如何通过在最后一个元素处拆分数组来将一维数组拆分为 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])

sample 运行-

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