我怎样才能使蟒蛇图的点随着时间的推移出现?

时间:2015-10-22 07:09:31

标签: python matplotlib

我想创建一个动画,我的数据点将逐渐显示在我的图表上,并在所有数据点出现时冻结。我已经看到完成了相关性,我只是不太确定如何用自己的个别点来做到这一点

这不会显示任何特别有用的东西,但我会觉得很酷,因为我试图在地图上可视化一些位置数据

我知道这不是很清楚所以请澄清,我不太清楚如何很好地表达我的问题。

谢谢

1 个答案:

答案 0 :(得分:9)

matplotlib.animation.FuncAnimation是适合您的工具。首先创建一个空图,然后在函数中逐渐添加数据点。以下代码将说明它:

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

x = np.arange(10)
y = np.random.random(10)

fig = plt.figure()
plt.xlim(0, 10)
plt.ylim(0, 1)
graph, = plt.plot([], [], 'o')

def animate(i):
    graph.set_data(x[:i+1], y[:i+1])
    return graph

ani = FuncAnimation(fig, animate, frames=10, interval=200)
plt.show()

结果(保存为gif文件)如下所示: enter image description here

编辑:要在matplotlib窗口中完成动画外观停止,您需要使其无限(在frames中省略FuncAnimation参数),并设置框架与框架系列中的最后一个数字对应:

def animate(i):
    if i > 9:
        i = 9
    graph.set_data(x[:i+1], y[:i+1])
    return graph

ani = FuncAnimation(fig, animate, interval=200)

或者,更好的是,您可以根据this问题的答案,将repeat中的FuncAnimation参数设置为False

编辑2:要为散点图设置动画,您需要一大堆其他方法。一段代码胜过千言万语:

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

x = np.arange(10)
y = np.random.random(10)
size = np.random.randint(150, size=10)
colors = np.random.choice(["r", "g", "b"], size=10)

fig = plt.figure()
plt.xlim(0, 10)
plt.ylim(0, 1)
graph = plt.scatter([], [])

def animate(i):
    graph.set_offsets(np.vstack((x[:i+1], y[:i+1])).T)
    graph.set_sizes(size[:i+1])
    graph.set_facecolors(colors[:i+1])
    return graph

ani = FuncAnimation(fig, animate, repeat=False, interval=200)
plt.show()