如何在Matplotlib中为动画的图像列表添加绘图?

时间:2018-05-03 16:00:59

标签: python animation matplotlib

我从here获取了以下代码。在plt.imshow()之后我想为每个“框架”绘制一个位于不同位置的红点。

如何使用plt.plot()唯一的plt.imshow()

下面的代码保留了先前从前面的“帧”中绘制的红点。

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

fig = plt.figure()


def f(x, y):
    return np.sin(x) + np.cos(y)

x = np.linspace(0, 2 * np.pi, 120)
y = np.linspace(0, 2 * np.pi, 100).reshape(-1, 1)
# ims is a list of lists, each row is a list of artists to draw in the
# current frame; here we are just animating one artist, the image, in
# each frame
ims = []
for i in range(60):
    x += np.pi / 15.
    y += np.pi / 20.
    im = plt.imshow(f(x, y), animated=True)

    # my added code here
    plt.plot(pos_x, pos_y, 'ro') # pos_x, pos_y change on every iteration

    ims.append([im])

ani = animation.ArtistAnimation(fig, ims, interval=50, blit=True,
                                repeat_delay=1000)

# ani.save('dynamic_images.mp4')

plt.show()

1 个答案:

答案 0 :(得分:1)

在动画对象列表中包含Line2D创建的plot对象。

line, = plt.plot(pos_x, pos_y, 'ro')

ims.append([im, line])