快捷導(dǎo)航

使用matplotlib動(dòng)態(tài)刷新指定曲線實(shí)例

更新時(shí)間：2020年04月23日 11:09:39 作者：LoveWeeknd

這篇文章主要介紹了使用matplotlib動(dòng)態(tài)刷新指定曲線實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

我就廢話不多說(shuō)啦，還是直接看代碼吧！

from matplotlib import pyplot as plt
import numpy as np
 
x = np.linspace(1, 100, 20)
y = x *2 +3
fig = plt.figure()
ax = fig.add_subplot(1,1,1)
ax.scatter(x, y)
plt.ion()
for i in range(10):
 y = x*i*0.1 + i
 try:
  ax.lines.remove(lines[0])
 except Exception:
  pass
 lines = ax.plot(x ,y)
 plt.pause(0.1)

補(bǔ)充知識(shí)：用Python的matplotlib庫(kù)動(dòng)態(tài)顯示不斷增長(zhǎng)的數(shù)據(jù)

"""
Created on Mon Dec 07 16:34:10 2015

@author: SuperWang
"""

import matplotlib.pyplot as plt
import numpy as np

fig,ax=plt.subplots()
fig2,ax2=plt.subplots()


y1=[]
y2=[]

for i in range(50):
 y1.append(np.sin(i))
 y2.append(np.cos(i))
 ax.cla()
 ax.set_title("Loss")
 ax.set_xlabel("Iteration")
 ax.set_ylabel("Loss")
 ax.set_xlim(0,55)
 ax.set_ylim(-1,1)
 ax.grid()
 ax.plot(y1,label='train')
 ax.plot(y2,label='test')
 ax.legend(loc='best')

 ax2.cla()
 ax2.set_title("Loss")
 ax2.set_xlabel("Iteration")
 ax2.set_ylabel("Loss")
 ax2.set_xlim(0,55)
 ax2.set_ylim(-1,1)
 ax2.grid()
 ax2.plot(y1,label='train')
 ax2.plot(y2,label='test')
 ax2.legend(loc='best')
 plt.pause(1)

要解決的問(wèn)題如標(biāo)題所示，原理很簡(jiǎn)單，就是當(dāng)數(shù)據(jù)增長(zhǎng)時(shí)，不斷清空以前的繪畫(huà)內(nèi)容，然后把現(xiàn)有的數(shù)據(jù)重新畫(huà)出來(lái)（數(shù)據(jù)是胡亂生成的）。

具體過(guò)程如下：

fig,ax=plt.subplots() 產(chǎn)生一個(gè)figure對(duì)象和一個(gè)axis對(duì)象。figure相當(dāng)于一個(gè)窗口，而axis相當(dāng)于一個(gè)畫(huà)布。此句也可以用兩句生成，即fig=plt.figure(num)，括號(hào)中的參數(shù)是figure的ID，如果只需創(chuàng)建一個(gè)figure對(duì)象，那么可以省略。然后ax=fig.subplot(1,1,1)，subplot()的具體用法可以去google或百度一下。ax.cla()就是在新數(shù)據(jù)到來(lái)時(shí)，先把之前的繪制的內(nèi)容清空，接下來(lái)，ax.set_title(“Loss”)，ax.set_xlabel(“Iteration”)，ax.set_ylabel(“Loss”)都很簡(jiǎn)單，見(jiàn)名知意。ax.set_xlim(0,55)，ax.set_ylim(-1,1)分別用來(lái)設(shè)置x軸和y軸的兩個(gè)端點(diǎn)。ax.grid()給畫(huà)布加上網(wǎng)格。ax.plot(y1,label='train')，ax.plot(y2,label='test')這兩句是實(shí)際的繪制命令，其中，參數(shù)label是為以后生成圖例用的。ax2.legend(loc='best')用來(lái)生成圖例，loc參數(shù)代表圖例位置location，而value：‘best'是其中的一種選擇，除此之外，還有左上角等其他選項(xiàng)。最后，plt.pause(1)是為了顯示上更直觀，故意每繪制一次，暫停1秒，注意，這里的單位是秒。如果是實(shí)際的應(yīng)用，而數(shù)據(jù)生成的過(guò)程又比較慢，此句完全可以省略。

這段代碼中創(chuàng)建了兩個(gè)窗口，在實(shí)驗(yàn)過(guò)程中，我發(fā)現(xiàn)只能有一個(gè)窗口被選中，即用鼠標(biāo)點(diǎn)擊哪個(gè)窗口，哪個(gè)窗口會(huì)動(dòng)態(tài)地顯示繪畫(huà)過(guò)程，而另一個(gè)保持不動(dòng)。

繪畫(huà)過(guò)程截圖如下：