python - 将图表与使用turtle模块制作的绘图结合起来

Question

我需要在绘制图形的同一画布上使用 turtle 模块绘制一些东西。

我曾考虑过使用 matplotlib 制作图表，但我需要稍后在该图表上用海龟绘制一些东西。我可以用海龟绘制图形和其他东西，但这要复杂得多。

谁能想到更好的方法？

标签： pythonmatplotlibdrawingturtle-graphics

Answer 1

既然你问了，这个问题就一直在我的脑海里。我研究了各种涉及堆叠隐形画布的方法，但没有运气。

我最终决定尝试在幕后进行绘图，将该绘图的图像保存到内存文件，然后将该图像加载到海龟背景中。FigureCanvasTkAgg通过引入 tkinter然后需要嵌入 tkinter 的海龟，它变得有点复杂。但它有效。

下面的代码使用 matplotlib 绘制一个正弦波，然后使用 turtle 在其上涂抹涂鸦 Hilbert 曲线。只是因为我可以：

import tkinter as tk
from io import BytesIO
from turtle import TurtleScreen, RawTurtle
from PIL import Image, ImageTk
import numpy as np

from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
from matplotlib.figure import Figure

root = tk.Tk()

# Plot graph
figure = Figure(figsize=(5, 5))
subplot = figure.add_subplot(111)
x = np.arange(0, 3 * np.pi, 0.1)
y = np.sin(x)
subplot.plot(x, y)

# Make memory image of graph
invisible_figure_canvas = FigureCanvasTkAgg(figure, root)
buffer = BytesIO()
figure.savefig(buffer, format="png")
buffer.seek(0)

# Open memory as tkinter image
image = Image.open(buffer)
photoImage = ImageTk.PhotoImage(image)
buffer.close()

# Now do our turtle drawing embedded in tkinter
canvas = tk.Canvas(root, width=500, height=500)
canvas.pack()

screen = TurtleScreen(canvas)
screen._setbgpic(screen._bgpic, photoImage)  # bypass restrictions (protected access)

turtle = RawTurtle(screen, shape='turtle')
turtle.shapesize(0.5)

def hilbertCurve(n, angle):
    if n <= 0:
        return

    turtle.left(angle)
    hilbertCurve(n - 1, -angle)
    turtle.forward(10)
    turtle.right(angle)
    hilbertCurve(n - 1, angle)
    turtle.forward(10)
    hilbertCurve(n - 1, angle)
    turtle.right(angle)
    turtle.forward(10)
    hilbertCurve(n - 1, -angle)
    turtle.left(angle)

hilbertCurve(4, 90)

screen.mainloop()