Помогите решить задачу по программированию пайтон

Для решения этой задачи вам нужно создать программу на Python, которая принимает ввод данных о дистанциях каждого этапа забега и номер этапа Джимми, после чего выводит дистанцию соответствующего этапа.

Вот пример решения:
def main():
# Ввод дистанций каждого этапа забега
distances = input("Введите дистанции каждого этапа через пробел: ").split()
# Конвертирование каждой дистанции в целое число
distances = [int(x) for x in distances]

# Ввод номера этапа Джимми
stage_number = int(input("Введите номер этапа Джимми: "))

# Вывод дистанции соответствующего этапа
print("Дистанция этапа {}: {}".format(stage_number, distances[stage_number - 1]))

if __name__ == "__main__":
main()

Этот код сначала запрашивает у пользователя дистанции каждого этапа забега, разделенные пробелом. Затем пользователь вводит номер этапа Джимми. Программа выводит дистанцию соответствующего этапа.