本次用Turtle模块实现Python二叉树

二叉树可以用递归来绘制

首先,导入模块

import turtle

设置每个树枝的倾斜角度

de=10

定义二叉树tree函数

def tree(x,y,degree,length,n):
    if n<=5: # 树叶
        t.pencolor("green")
    else: # 树枝
        t.pencolor("brown")
    t.pu()
    t.goto(x,y)
    t.setheading(degree)
    t.pd()
    t.forward(length)
    if n==1:
        return
    else:
        pos=t.pos()
        tree(pos[0],pos[1],degree+de,length,n-1) # 向左延申的树枝
        tree(pos[0],pos[1],degree-de,length,n-1) # 向右延申的树枝

设置turtle画笔参数

t.speed(0) # 速度加快
t.delay(0) # 由于设置为最高速度后速度依然很慢,画布设置为无延迟
t.tracer(400) # 依然很慢。。。加速400倍
tree(0,-300,90,30,10) # 画二叉树
t.done() # 窗口保持显示

最终代码

import turtle as t

de=10 # 每根树枝的倾斜角度

def tree(x,y,degree,length,n):
    if n<=5:
        t.pencolor("green")
    else:
        t.pencolor("brown")
    t.pu()
    t.goto(x,y)
    t.setheading(degree)
    t.pd()
    t.forward(length)
    if n==1:
        return
    else:
        pos=t.pos()
        tree(pos[0],pos[1],degree+de,length,n-1)
        tree(pos[0],pos[1],degree-de,length,n-1)

t.speed(0)
t.delay(0)
t.tracer(400)
tree(0,-300,90,30,10)
t.done()

让二叉树的倾斜角度、长度和数量变得随机

导入random

import random

将函数改写为

def tree(x,y,degree,length,n):
    if n<=5:
        t.pencolor("green")
    else:
        t.pencolor("brown")
    t.pu()
    t.goto(x,y)
    t.setheading(degree)
    t.pd()
    t.forward(length)
    if n==1:
        return
    else:
        pos=t.pos()
        tree(pos[0],pos[1],degree+de+random.randint(-5,5),length+random.randint(-5,5),n-1 if random.randint(1,100)<=90 else n)
        tree(pos[0],pos[1],degree-de+random.randint(-5,5),length+random.randint(-5,5),n-1 if random.randint(1,100)<=90 else n)

最后两行:倾斜角偏移-5~5度,长度偏移-5~5,树枝10%的可能增加1条

效果:


制作不易,喜欢的话记得关注哦!

更多推荐