【Python】Trutle画变色蟒蛇

import turtle

def changed_color(i, r, g, b):
    if i==2:
        r, g, b = 0, 160, 0
    turtle.pencolor(r, g, b)
    return r, g, b

def draw_snake(radius, angle, length, r, g, b):
    for i in range(length):
        r, g, b = changed_color(i, r, g, b)
        g = g + 20
        turtle.circle(radius, angle)
        r, g, b = changed_color(i, r, g, b)
        turtle.circle(-radius, angle)
    turtle.circle(radius, angle/2)
    changed_color(0, 160, 102, 240)
    turtle.fd(40)
    changed_color(0, 160, 62, 240)
    turtle.circle(16, 180)
    turtle.fd(40 * 2 / 3)

def main():
    turtle.setup(650, 350, 0, 0)
    turtle.colormode(255)
    turtle.penup()
    turtle.fd(-250)
    turtle.pendown()
    turtle.pensize(25)
    turtle.seth(-40)
    r, g, b = 0, 100, 0
    draw_snake(40, 80, 4, r, g, b)
    turtle.done()

if __name__ == '__main__':
    main()