Matplotlib图中的进度线

时间:2012-08-13 21:57:24

标签: python matplotlib

目标:绘制图形(x,y)并将图形w.r.t上的垂直线移动到计时器。

我开始使用matplotlib实现这一点。可以使用matplotlib的draw()功能来实现它,但它每次都会重新绘制cpu,因为它不允许我与图形交互。所以我决定使用matplotlib的动画功能。在未来,我还想暂停移动线。所以我不能使用matplotlib.animation.FuncAnimatin

问题:我使用canvas.copy_from_bbox(ax.bbox),ax.draw_artist(line),canvas.blit(ax.bbox)。 但是,我无法将图形存储在背景中并在其上移动一条线。当我尝试存储时,它会以一种非常奇怪的方式覆盖。

这是我建立的代码。有人可以帮帮我吗?提前致谢。

import sys
import matplotlib.pyplot as p
import time
fig=p.figure();
ax = fig.add_subplot(1,1,1)

y=[];x=[];y1=[0,1000];x1=[0,0]
y=numpy.random.randn(1000,1)*100
x=numpy.arange(0,1000)
line1, = ax.plot(x,y,color='black');
ax.set_ylim(0, 1000);
line, = ax.plot(x1,y1,color='r',alpha=1,animated=True); # this is the line which i wanted to move over the graph w.r.t to time. ( i can also use axvline , but i guess its the same).
canvas = ax.figure.canvas
canvas.draw()
background = canvas.copy_from_bbox(ax.bbox); #my problem is here
starttime=time.time();
mytimer=0;
mytimer_ref=0;
def update(canvas,line,ax):
    canvas.restore_region(background) #my problem is here 
    t=time.time()-starttime;
    mytimer=t+mytimer_ref;
    x1=[mytimer,mytimer];
    line.set_xdata(x1);
    ax.draw_artist(line)
    canvas.blit(ax.bbox) #my problem is here

def onclick(event):
    global starttime
    starttime=time.time();
    global mytimer_ref;
    mytimer_ref=event.xdata;
    print "starttime",starttime;


cid=line1.figure.canvas.mpl_connect('button_press_event',onclick); # when i click the mouse over a point, line goes to that point and start moving from there. 
timer=fig.canvas.new_timer(interval=100);
args=[canvas,line,ax];
timer.add_callback(update,*args); # every 100ms it calls update function
timer.start();
p.show();

1 个答案:

答案 0 :(得分:4)

所以看起来你所指的“非常奇怪的方式”实际上是用你的background = canvas.copy_from_bbox(ax.bbox)捕获了错误的bbox。我相信这是大多数后端的已知问题,其中添加工具栏等会影响bbox的位置。

基本上,如果你可以在之后捕捉背景窗口弹出,那么一切都应该适合你。这可以通过多种方式完成,在您的情况下,最简单的方法是将canvas.draw()命令替换为plt.show(block=False),这将调出窗口,而不会使其成为阻塞命令。

稍微补充一下,我确定你知道在python代码中不需要分号,但是在我调试的时候,我稍微整理了你的代码(没有完全结束):

import sys
import matplotlib.pyplot as plt
import time
import numpy


fig = plt.figure()
ax = fig.add_subplot(111)


max_height = 100
n_pts = 100
y1 = [0, max_height]
x1 = [0, 0]
y = numpy.random.randn(n_pts) * max_height
x = numpy.arange(0, n_pts)

# draw the data
line1, = ax.plot(x, y, color='black')

# fix the limits of the plot
ax.set_ylim(0, max_height)
ax.set_xlim(0, n_pts)

# draw the plot so that we can capture the background and then use blitting
plt.show(block=False)

# get the canvas object
canvas = ax.figure.canvas
background = canvas.copy_from_bbox(ax.bbox)

# add the progress line.
# XXX consider using axvline
line, = ax.plot(x1, y1, color='r', animated=True) 


starttime=time.time()
mytimer=0
mytimer_ref=0

def update(canvas, line, ax):
    # revert the canvas to the state before any progress line was drawn
    canvas.restore_region(background)

    # compute the distance that the progress line has made (based on running time) 
    t = time.time() - starttime
    mytimer = t + mytimer_ref
    x1 = [mytimer,mytimer]
    # update the progress line with its new position
    line.set_xdata(x1)
    # draw the line, and blit the axes
    ax.draw_artist(line)
    canvas.blit(ax.bbox)

def onclick(event):
    global starttime
    starttime=time.time()
    global mytimer_ref
    mytimer_ref=event.xdata
    print "starttime",starttime


cid=line1.figure.canvas.mpl_connect('button_press_event',onclick) # when i click the mouse over a point, line goes to that point and start moving from there. 
timer=fig.canvas.new_timer(interval=100)
args=[canvas,line,ax]
timer.add_callback(update,*args) # every 100ms it calls update function
timer.start()
plt.show()

HTH