matplotlib可以說是python繪圖中最最最經(jīng)典的庫，然而，當(dāng)數(shù)據(jù)源較大時，其保存速度也會變得極慢，苦其久矣，其它繪圖庫，要么資料極少，要么不好用，那么，只能從其本身出發(fā)，想辦法優(yōu)化保存速度了。

在嘗試了不限于jit、清除畫布等等方法均失敗后，終于探索出一種有效的辦法！

保存時間從3.2s提升到了1.6s！??！

解決辦法

• 找到你虛擬環(huán)境中 matplotlib 里pyplot.py的位置

例：我的是在 D:\CRR\venv\Lib\site-packages\matplotlib\pyplot.py

一般都是在你項(xiàng)目目錄的 venv\Lib\site-packages\matplotlib\pyplot.py

• 打開此腳本，找到 savefig 里的 fig.canvas.draw_idle()，將其注釋，我的是在978行，行號不一定相同

效果

這里是注釋了的，保存速度為1.6s

這里是沒注釋的，速度為3.2s

注意事項(xiàng)

如果你需要不斷去plt.show()，不斷刷新、渲染，不推薦此方法（我沒這個需求，還沒測試過此法是否試用這種情況）

如果你僅需保存圖片到本地（哪怕需要背景透明（親測）），直接使用此方法即可?。?！

考慮到直接修改了庫的源碼，為避免其它場景下的沖突，建議在修改源碼時，加個參數(shù)控制是否啟用那行命令

分析過程

1.在pycharm里，使用這種方法分析運(yùn)行情況

按照運(yùn)行時間排序，發(fā)現(xiàn)在保存這一步花了3.6s的時間，就一行就這么慢?。?！

plt.savefig(savepath, pad_inches=0.0, transparent=True)

2.雙擊點(diǎn)開后

發(fā)現(xiàn)在savefig后還有一步

fig.canvas.draw_idle() # need this if ‘transparent=True' to reset colors

注釋里我大概理解是指：當(dāng)設(shè)置背景透明且需要刷新顏色時需要這一步 又看了看這句話消耗的時間，占了保存時的一半?。?！官方還貼心的備注了可以根據(jù)具體情況重寫此方法

而我們僅需保存，并沒有重新渲染的需求

3.而且官方的注釋里說了，這里就是針對GUI時間循環(huán)時刷新小部件的，我單純的后端保存，完全用不到才對，

4.嘗試將其注釋，看是否會影響到繪圖結(jié)果

經(jīng)測試，并無影響?。?！包括尺寸、大小 完全一樣?。?！

其實(shí)還是推薦，重寫一下savefig，加一個參數(shù)來判斷是否關(guān)閉這個draw_idle，省的影響到后續(xù)其它的使用，畢竟是直接注釋了源碼

總結(jié)

以上為個人經(jīng)驗(yàn)，希望能給大家一個參考，也希望大家多多支持腳本之家。

最新評論