VSplit Simple

import numpy as np

a = np.arange(10).reshape(2, 5)

a

array([[0, 1, 2, 3, 4],
       [5, 6, 7, 8, 9]])

x, y = np.vsplit(x, 2)

x

array([[0, 1, 2, 3, 4]])

y

array([[5, 6, 7, 8, 9]])

a.shape

(2, 5)

x.shape

(1, 5)

y.shape

(1, 5)