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

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

保存時(shí)間從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

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

效果

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

這里是沒(méi)注釋的，速度為3.2s

注意事項(xiàng)

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

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

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

分析過(guò)程

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

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

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

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

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

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

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

而我們僅需保存，并沒(méi)有重新渲染的需求

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

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

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

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

總結(jié)

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

最新評(píng)論