Как найти смежные области в 2D данных

Question

Я хочу пометить несмежные области на изображении разными метками. Я думаю, это должно быть возможно с scikit-learn. Изображение хранится, например, в двумерном numpy.ndarray с нулями и единицами, которые имеют разные смежные области.

Давайте посмотрим на следующий очень простой массив, в котором у нас есть две смежные области единиц, но этичтобы регионы были отделены друг от друга нулями.

np.array([
        [1, 1, 1, 0, 0, 0],
        [1, 1, 0, 0, 0, 1],
        [0, 1, 0, 1, 0, 1],
        [1, 1, 0, 1, 1, 1]
], dtype = int)

алгоритм должен пометить верхнюю левую непрерывную область меткой, например «1», а правую нижнюю непрерывную область второй меткой «2».

np.array([
        [1, 1, 1, 0, 0, 0],
        [1, 1, 0, 0, 0, 2],
        [0, 1, 0, 2, 0, 2],
        [1, 1, 0, 2, 2, 2]
], dtype = int)

Любые советы, как это сделать?