如this question中所述,我试图在iPython笔记本中(在一个单元格中)动态更新绘图。 不同之处在于我不想绘制新行,但是我的x_data和y_data在某个循环的每次迭代中都在增长。
我想做的是:
import numpy as np
import time
plt.axis([0, 10, 0, 100]) # supoose I know what the limits are going to be
plt.ion()
plt.show()
x = []
y = []
for i in range(10):
x = np.append(x, i)
y = np.append(y, i**2)
# update the plot so that it shows y as a function of x
time.sleep(0.5)
但是我希望情节有一个传奇,如果我这样做
from IPython import display
import time
import numpy as np
plt.axis([0, 10, 0, 100]) # supoose I know what the limits are going to be
plt.ion()
plt.show()
x = []
y = []
for i in range(10):
x = np.append(x, i)
y = np.append(y, i**2)
plt.plot(x, y, label="test")
display.clear_output(wait=True)
display.display(plt.gcf())
time.sleep(0.3)
plt.legend()
我最终得到一个包含10个项目的图例。如果我将plt.legend()
放在循环中,则每次迭代都会增加图例...任何解决方案?
答案 0 :(得分:6)
目前,您每次在循环中plt.plot
创建一个新的Axes对象。
因此,如果在使用plt.gca().cla()
之前清除当前轴(plt.plot
),并将图例放在循环中,则每次都不会增加图例:
import numpy as np
import time
from IPython import display
x = []
y = []
for i in range(10):
x = np.append(x, i)
y = np.append(y, i**2)
plt.gca().cla()
plt.plot(x,y,label='test')
plt.legend()
display.clear_output(wait=True)
display.display(plt.gcf())
time.sleep(0.5)
修改强>
正如@tcaswell在评论中指出的那样,使用%matplotlib notebook
magic命令为您提供了可以更新和重绘的实时图形。