import numpy as np
a = np.array([0, 0, 1, 1, 1, 3, 5, 5, 5])
from itertools import groupby
result = np.array([list(g) for _, g in groupby(a)])
как переместить одинаковые элементы в массиве numpy в подмассивы