PyQt5 matplotlib畫圖不刷新的解決方案

Posted on by

本人最近在做一個GUI項目時，需要用到matplotlib畫圖，寫完代碼運行時，發現所需要畫圖功能不能正常使用，在觸發畫圖事件後，畫佈未更新，需要放大界面才能更新（奇葩吧）。

經過資料查閱，發現需要畫圖函數後面添加

self.fig.canvas.draw_idle()

補充：解決PyQt5中使用柱狀圖和餅狀圖刷新問題

一、問題描述

在做一款番茄鐘應用，其中包含數據統計功能，如下圖：

我遇到的問題是：當完成一個番茄鐘後，需要對統計數據進行刷新，但是重新調用一下畫圖函數無效，下面貼出調用的這個畫圖函數：

class MyFigure(FigureCanvas, QWidget):
 def __init__(self,parent=None, minWidth=600, minHeight=380, dpi=120):
  #第一步：創建一個創建Figure
  self.fig = Figure(figsize=(8, 8), dpi=dpi)
  #第二步：在父類中**Figure窗口
  super(MyFigure,self).__init__(self.fig) #此句必不可少，否則不能顯示圖形
  #第三步：創建一個子圖，用於繪制圖形用，111表示子圖編號，如matlab的subplot(1,1,1)
  self.axes = self.fig.add_subplot(111)
  self.setMinimumSize(minWidth, minHeight)
  FigureCanvas.updateGeometry(self)
 #第四步：就是畫圖，【可以在此類中畫，也可以在其它類中畫】
 #畫日分佈圖 
 def drawDayChart(self):
  periodHour=['0'+str(i) if i<10 else str(i) for i in range(24)]
  periodDict=dict(zip(periodHour, [0]*24))
  with open('configFiles/history.txt', 'r') as f:
   lines=f.readlines()
   for line in lines:
    period=line[11:13]
    periodDict[period]+=1
  periodDictKeys=periodDict.keys()
  periodDictValues=periodDict.values()
  rects=self.axes.bar(periodDictKeys, periodDictValues, align='edge', width=0.9, color='green')
  self.axes.set_ylabel('番茄鐘個數')
  self.axes.set_title('番茄鐘日分佈', color='blue')
  for rect in rects:
   x=rect.get_x()
   y=rect.get_height()
   if(y==0):
    continue
   self.axes.text(x, 1.01*y, str(y), va='bottom')

二、解決思路

1、首先確定每次刷新的時候，穿給圖標的數據是否真正刷新，確定這一點後

2、嘗試使用self.axes.clear()將之前畫的圖刪除掉，然後重畫—嘗試後仍然不行

3、google :How to update a plot in matplotlib?

找到瞭這麼幾行代碼：

 fig.canvas.draw()
 fig.canvas.flush_events()

於是去matplotlib官網查文檔，在我的函數中加入瞭self.fig.canvas.draw_idle()。

4、成功實現瞭刷新

最後的代碼：

class MyFigure(FigureCanvas, QWidget):
 def __init__(self,parent=None, minWidth=600, minHeight=380, dpi=120):
  #第一步：創建一個創建Figure
  self.fig = Figure(figsize=(8, 8), dpi=dpi)
  #第二步：在父類中**Figure窗口
  super(MyFigure,self).__init__(self.fig) #此句必不可少，否則不能顯示圖形
  #第三步：創建一個子圖，用於繪制圖形用，111表示子圖編號，如matlab的subplot(1,1,1)
  self.axes = self.fig.add_subplot(111)
  self.setMinimumSize(minWidth, minHeight)
  FigureCanvas.updateGeometry(self)
 #第四步：就是畫圖，【可以在此類中畫，也可以在其它類中畫】
 #畫日分佈圖 
 def drawDayChart(self):
  periodHour=['0'+str(i) if i<10 else str(i) for i in range(24)]
  periodDict=dict(zip(periodHour, [0]*24))
  with open('configFiles/history.txt', 'r') as f:
   lines=f.readlines()
   for line in lines:
    period=line[11:13]
    periodDict[period]+=1
  periodDictKeys=periodDict.keys()
  periodDictValues=periodDict.values()
  #實現刷新的條件1
  self.axes.clear()
  rects=self.axes.bar(periodDictKeys, periodDictValues, align='edge', width=0.9, color='green')
  #實現刷新的條件2
  self.fig.canvas.draw_idle()
  self.axes.set_ylabel('番茄鐘個數')
  self.axes.set_title('番茄鐘日分佈', color='blue')
  for rect in rects:
   x=rect.get_x()
   y=rect.get_height()
   if(y==0):
    continue
   self.axes.text(x, 1.01*y, str(y), va='bottom')

三、原因猜測

即使清空瞭self.axes 但是畫佈中可能還存在殘留數據

self.axes.fig.draw_idle() 在官方文檔中的描述是用於重畫圖表，因此嘗試使用，最終解決瞭問題！

以上為個人經驗，希望能給大傢一個參考，也希望大傢多多支持WalkonNet。如有錯誤或未考慮完全的地方，望不吝賜教。

推薦閱讀：