画彩色的二叉树

#把树枝变成彩色/渐变色2000/10/14
import turtle

def draw_branch(branch_length):
    turtle.speed(10)
    if branch_length>5:
        #绘制右侧树枝
        turtle.forward(branch_length)
        turtle.right(20)
        #每次树枝变短15
        draw_branch(branch_length-15)
        #绘制左侧树枝
        turtle.left(40)
        draw_branch(branch_length-15)
        #返回之前的节点处
        turtle.right(20)
        if branch_length <= 15:  # 树枝长小于30，可以当作树叶了，树叶部分为绿色
            turtle.pencolor('green')
        if branch_length <= 30 and branch_length>15:
            turtle.pencolor('brown')  # 树干部分为棕色
        if branch_length <= 45 and branch_length>30:
            turtle.pencolor('blue')  # 树干部分为蓝色
        if branch_length <= 60 and branch_length>45:
            turtle.pencolor('yellow')  # 树干部分为黄色
        if branch_length <= 75 and branch_length>60:
            turtle.pencolor('red')  # 树干部分为红色
        if branch_length <= 90 and branch_length>75:
            turtle.pencolor('purple')  # 树干部分为紫色
        turtle.backward(branch_length)


def main():
    turtle.left(90)
    turtle.penup()
    #往下移150单位
    turtle.backward(150)
    turtle.pendown()
    #90=15*6,有6次分叉
    draw_branch(90)
    turtle.down()
    turtle.exitonclick()


if __name__=='__main__':
    main()