如何通过将数组在最后一个元素处分割来将NumPy的一维数组分割成二维数组？

Question

我有一个numpy数组，比如说

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

我希望将它分成一个二维数组，使得最后一个元素在其自己的数组中，像这样：

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

我应该如何做到这一点？

- HMK

---

顺便说一句，那不是一个二维数组，只是两个独立的一维数组或列表。 - Divakar

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

- Divakar