How to split 1D array into 2D array in NumPy by splitting the array at the last element?

Question

I have a numpy array, say

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

I want to split it into a 2d array such that the last element is in its own array, like this

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

How exactly would I do this?

BTW that's not a 2D array, just two separate 1D arrays or lists. — Divakar
– Divakar, Commented Feb 7, 2017 at 20:27

Divakar · Accepted Answer · 2017-02-07 19:39:40Z

5

np.split(a,[-1])

Sample run -

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

answered Feb 7, 2017 at 19:39

Divakar

222k19 gold badges273 silver badges374 bronze badges

Sign up to request clarification or add additional context in comments.

1 Answer 1