Как разделить массив так, чтобы получился массив?

Данные генерируются: test_arr = np.arange(55)

Функция делит массив на вектора:

def traincount(u, v): return [u[v:v+n] for v in range(0, len(u), n)] 

Получилось: [array([0,..9]), array([10,..19]),...]

А нужен массив [[0,..9], [10,..19],.. ]. Как получить желаемый результат?