Задача программирования "Размер реки"

Задача программирования "Размер реки" заключается в написании программы, которая принимает на вход двумерный массив, представляющий карту земельного участка, где каждый элемент массива может принимать значения 0 или 1. Здесь 0 обозначает сушу, а 1 - реку. Необходимо найти и вернуть массив с размерами всех рек на участке. Размер реки определяется количеством единиц (1) в непрерывной группе смежных элементов массива. Смежные элементы - это элементы, расположенные по горизонтали или вертикали (не по диагонали). Пример в видео Python