Всем добрый день с задачкой на python. Условие: Класс написал контрольную. Если все оценки одинаковые, учитель понимает, что ученики списали друг у друга и надо дать контрольную снова. Если разные, то не надо. Дана последовательность оценок и 0 в конце. Если все цифры в столбце одинаковые, выведи "ДА", иначе "НЕТ".
Ввод Вывод
5 ДА
5
5
5
5
0

uchenik1= int(input("Оценка ученика 1:"))

uchenik2= int(input("Оценка ученика 2:"))

uchenik3= int(input("Оценка ученика 3:"))

uchenik4= int(input("Оценка ученика 4:"))

uchenik5= int(input("Оценка ученика 5:"))

uchenik6= int(input("Оценка ученика 6:"))

if uchenik1 ==5 and  uchenik2 ==5 and uchenik3 ==5 and uchenik4 ==5 and uchenik5 ==5 and  uchenik6 ==5:

  print("Переписываем рабата")

else:

   print("Все молодцы")

Объяснение:

Хорошего дня ))

marks = []

while True:

   mark = int(input('Введите оценку: '))

   if mark == 0:

       break

   marks.append(mark)

if len(marks) == marks.count(marks[0]):

   print('ДА')

else:

   print('НЕТ')