Вопрос Задание на питон напишите ответ

Назовём натуральное шестизначное число N (100000≤ N ≤ 999999) счастливым, если суммы трёх его первых и трёх последних цифр различаются не более, чем на 3. Найдите количество таких чисел.

def is_lucky(number):
# Разделяем число на первые и последние три цифры
first_three = number // 1000
last_three = number % 1000

# Сумма первых трёх цифр
sum_first = sum(map(int, str(first_three)))
# Сумма последних трёх цифр
sum_last = sum(map(int, str(last_three)))

# Проверяем условие
return abs(sum_first - sum_last) <= 3

def count_lucky_numbers():
count = 0
for number in range(100000, 1000000): # От 100000 до 999999
if is_lucky(number):
count += 1
return count

# Вывод результата
print("Количество счастливых чисел:", count_lucky_numbers())

В каком виде должен быть ответ (функция, скрипт, класс)?

Тебе код написать или просто ответ нужен?

Количество счастливых шестизначных чисел равно 668304