Odpowiedź
Napisałam kod z tymi zmianami, które objaśnił dfw.
Dodatkowo wstawiłam spacje tam gdzie powinny być. Nie tylko dlatego, że ktoś tak kazał, ale przede wszystkim dlatego, że tak się wygodniej czyta!
Dodatkowo kod można uprościć, ponieważ wartość zmiennej wynik jest zawsze wypisywana, więc print() może być jeden raz poza obszarem if/else.