определите сколько раз в данное четырехзначное число входит цифра 1.Питон

num = int(input("n = "))

ones = 0

for i in range(4):

   if num % 10 == 1:

       ones += 1

   num //= 10

print(ones)