Matplotlib - Subplot中的2个数字 - 1是动画

时间:2012-11-13 17:57:01

标签: python animation matplotlib

我想在子图中绘制两个数字:

fig = plt.figure()
ax1 = fig.add_subplot(1,2,1)
ax2 = fig.add_subplot(1,2,2)

假设ax1将填充添加点的动画(散点图)。然后,Ax2将这些点分成网格,并显示密度。

我可以在子图1中显示动画,并在完成时将密度图像添加到subplot2吗?

1 个答案:

答案 0 :(得分:2)

这应该是可能的。请查看example。您也可以查看上一个问题:

Simple animation of 2D coordinates using matplotlib and pyplot

以下是一个示例实现。第二个图是隐藏的,直到第一个停止渲染:

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

def update_line(num, data, line, img):
    line.set_data(data[...,:num])
    if num == 24:
        img.set_visible(True)
    return line, img

fig1 = plt.figure()

data = np.random.rand(2, 25)
ax1=plt.subplot(211)
l, = plt.plot([], [], 'rx')
plt.xlim(0, 1)
plt.ylim(0, 1)
plt.xlabel('x')
plt.title('test')
ax2=plt.subplot(212)
nhist, xedges, yedges = np.histogram2d(data[0,:], data[1,:])
img = plt.imshow(nhist, aspect='auto', origin='lower')
img.set_visible(False)
line_ani = animation.FuncAnimation(fig1, update_line, 25, 
                                   fargs=(data, l, img),
                                   interval=50, blit=True)
line_ani.repeat = False
plt.show()