Единицы и нули

Кодинг-марафон. Задача № 5.

Напишите функцию, которая возвращает True, если в переданном числе за каждой последовательностью единиц следует последовательность нулей той же длины.

Примеры

same_length (110011100010) ➞ True

same_length (101010110) ➞ False

same_length (111100001100) ➞ True

same_length (111) ➞ False

Решение

def same_length(number):
    num_as_string
    while '10' in num_as_string:
        num_as_string = num_as_string.replace('10', '')
    return len(num_as_string) == 0