Напишите функцию output(year), которая для заданного года year (тип int) возвращает чего было больше воскресений или сред в этом году. Функция должна выдавать: 0 - если было больше воскресений; 1 - если больше сред; 2 - если количество воскресений такое же, как и количество сред. (Python)
year = int(input("Напишите год"))
def YearSorW():
day = year*365
W = day/7
S = day/7
if W < S:
output = 0
elif W > S:
output = 1
else:
output = 2
return output
print(YearSorW())