matplotlib 如何更新形状

bogh5gae 于 2023-10-24 发布在其他

关注(0)|答案(1)|浏览(110)

我有一个矩形的形状是在一个图像上绘制。我想移动的形状，即改变其坐标，绘制新的形状，并删除旧的（背景图像保持不变）
类似于这个代码的东西。它显示了一个矩形三次，但我想当绿色形状被绘制时，红色的要被删除，同样的蓝色。当蓝色出现时，删除绿色。

import matplotlib.pyplot as plt
import numpy as np

# Get an example image
image_file = r"image_splash.jpg"
img = plt.imread(image_file)

# Make some example data
x = np.array([475, 475, 675, 675, 475])
y = np.array([100, 200, 200, 100, 100])

# Create a figure
_dpi = 72
fig = plt.figure(1, figsize=(1000/_dpi, 1000/_dpi), dpi=_dpi)
ax = fig.add_subplot(1, 1, 1)
ax.set_aspect('equal')

# Show the image
ax.imshow(img)

# Now, loop through the offset array and update the shape 
# (This obviously doesnt work, it plots all three shapes)
rgb = ['r', 'g', 'b']
offset = [0, 150, 300]
plt.plot(x, y + offset[0], c=rgb[0]) # draw the red rect
plt.plot(x, y + offset[1], c=rgb[1]) # should remove the red and draw the green
plt.plot(x, y + offset[2], c=rgb[2]) # should remove the green and draw the blue

matplotlib

来源：https://stackoverflow.com/questions/66464565/how-to-update-a-shape

1条答案

按热度按时间

cwdobuhd1#

您可以尝试matplotlib的动画API，如下所示

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

fig, ax = plt.subplots()
xdata, ydata = [], []
(ln,) = plt.plot([], [], "r")

x = np.array([475, 475, 675, 675, 475])
y = np.array([100, 200, 200, 100, 100])
offset = [0, 150, 300]
colors = {
    0: "red",
    150: "green",
    300: "blue",
}

def init():
    ax.set_xlim(0, 1000)
    ax.set_ylim(0, 1000)
    return (ln,)

def update(_offset):
    ln.set_data(x, y + _offset)
    ln.set_color(colors[_offset])
    return (ln,)

def gety():
    for i in range(1000):
        yield offset[i % 3]

ani = FuncAnimation(fig, update, frames=gety(), init_func=init, blit=True)
plt.show()

赞(0）回复(0）举报 2023-10-24

我来回答

matplotlib 如何更新形状

1条答案

相关问题

热门标签

最新问答