Question

我需要绘制很多线条，并且我想在计算时显示它们。代码看起来像这样：

x = arange(100000)
for y in range(100000):
    ax.plot(x*y)
    draw()

现在，正如您可以想象的那样，这会非常快地变慢。我想我可以做的是绘图，将绘图保存到缓冲区，清除绘图，将缓冲区放置为背景，然后绘制下一行。这样，我最终没有这么多Line2D对象。有人会有任何想法吗？

Answer 1

您似乎需要matplotlib.animation功能。 animation examples

编辑：添加了我自己版本的更简单的示例代码。

import random
from matplotlib import pyplot as plt
from matplotlib import animation

def data_generator(t):
    if t<100:
        return random.sample(range(100), 20)

def init():
    return plt.plot()

def animate(i):
    data = data_generator(i)
    return plt.plot(data, c='k')

fig = plt.figure()
anim = animation.FuncAnimation(fig, animate, init_func=init, frames=1000, interval=1000, blit=True)
plt.show()

EDIT2：多行实时更新版本。

import random
from matplotlib import pyplot as plt
from matplotlib import animation

def data_generator_1(t):
    if t<100:
        x1.append(t)
        y1.append(random.randint(1, 100))

def data_generator_2(t):
    if t<100:
        x2.append(t)
        y2.append(random.randint(1, 100))

def init():
    global x1
    global y1
    x1 = []
    y1 = []

    global x2
    global y2
    x2 = []
    y2 = []

    l1, l2 = plt.plot(x1, y1, x2, y2)
    return l1, l2

def animate(i):
    data_generator_1(i)
    data_generator_2(i)
    l1, l2 = plt.plot(x1, y1, x2, y2)
    plt.setp(l1, ls='--', c='k')
    plt.setp(l2, c='gray')
    return l1, l2

fig = plt.figure()
anim = animation.FuncAnimation(fig, animate, init_func=init, frames=1000, interval=1000, blit=True)
plt.show()

我很确定有很多方法可以在没有全局变量的情况下绘制这种动画。这只是快速试用，向您展示您想要的可能性。

我不知道你的第一条评论所认为的ipython / vanilla脚本问题。所有示例都在普通编辑器（而不是ipython）上编码。也许有matplotlib版本的差异。

Answer 2

或者，如果你正在使用ipython笔记本，你可以使用IPython显示功能：

from IPython import display
import matplotlib.pyplot as plt
import numpy as np
%matplotlib

x = np.arange(100)

for y in np.arange(100):  
    fig, ax = plt.subplots(1,1, figsize=(6,6))
    ax.plot(x * y)
    ax.set_ylim(0, 10000) # to keep the axes always the same
    display.clear_output(wait=True)
    display.display(fig)
    plt.close()

如果你想在任何时候说，10行一次画一个，你可以这样做：

x = np.arange(100)
fig, ax = plt.subplots(1,1, figsize=(6,6))
for y in np.arange(100):        
    ax.plot(x*y)
    ax.set_ylim(0,10000)
    display.clear_output(wait=True)
    display.display(fig)
    if y > 10:           # from the 10th iteration,
        ax.lines.pop(0)  # remove the first line, then the 2nd, etc.. 
                         # but it will always be in position `0`
plt.close()

HTH

动态过度绘图

2 个答案: