elements.count(elements[0]) != len(elements) is logic statement itself, therefore you can return it. Futhermore you can join both case in one: return elements == [] or elements.count(elements[0]) == len(elements) More