цвета закодированы следующим образом 1 красный,2 зелёный , 3 жёлтый. По введёному номеру цвета нарисовать произвольную окружность с заливкой этого цвета

import turtle; n=int(input()); c=['red','green','yellow'][n-1]; t=turtle.Turtle(); t.fillcolor(c); t.begin_fill(); t.circle(50); t.end_fill(); turtle.done()