大家好，本文將分享如何使用matplotlib制作動態(tài)條形圖，制作的圖很美，這個是我在之前發(fā)布的一篇中使用的圖片，

效果如下

制作思路

為了方便大家學(xué)習(xí)，我將不直接進行講解，而是以我是如何一步步制作的思路來介紹整個過程。

說到用 Python 制作動態(tài)圖，首先想到的肯定是一些直接拿來就用的庫，雖然我沒做過，但是我相信一定有且不止一個，搜了一圈后發(fā)現(xiàn)有個bar chart race庫看起來不錯

畢竟有現(xiàn)成的輪子，只需要填充數(shù)據(jù)即可，但是研究了一番，正如我之前所說，越好看的圖對數(shù)據(jù)的要求越高，這個動態(tài)圖的庫指定數(shù)據(jù)必須類似如下格式

即有時間和國家兩個維度，通過時間的變化來動態(tài)變化數(shù)據(jù)，但是我的數(shù)據(jù)很簡單，就一列人名一列數(shù)據(jù)如下

如果再將數(shù)據(jù)強行修改成該庫所需要的格式，花費的功夫?qū)⒏?！所以果斷放棄，回到我熟悉?code>matplotlib上，首先很自然的想法就是，根據(jù)每一行，寫一個循環(huán)，每次循環(huán)就創(chuàng)建一個條形圖。

這個想法是完全可以的，也有很多人這么做了，但是有一個問題就是做出來的效果太難看了，并且每次循環(huán)都需要重新生成畫布繪圖需要一定時間，這也會讓最后的動態(tài)圖顯的不連貫。所以稍作嘗試后這個思路也被拋棄了。

animation

又經(jīng)過一番探索后，終于找到一個合適的工具——animation模塊，可以完美的實現(xiàn)我的需求，只需要將你的繪圖函數(shù)寫好，將涉及到的參數(shù)空出來，通過FuncAnimation來更新數(shù)據(jù)即可。

當然我也是第一次接觸，雖了解思路但不清楚具體實現(xiàn)邏輯，所以先從官方拉一個demo來看看整個過程如何實現(xiàn)

fig, ax = plt.subplots()
xdata, ydata = [], []
ln, = ax.plot([], [], 'r-', animated=False)

def init():
    ax.set_xlim(0, 2*np.pi)
    ax.set_ylim(-1, 1)
    return ln,

def update(frame):
    xdata.append(frame)
    ydata.append(np.sin(frame))
    ln.set_data(xdata, ydata)
    return ln,

ani = FuncAnimation(fig, update, frames=np.linspace(0, 2*np.pi, 128),
                    init_func=init, blit=True)

HTML(ani.to_jshtml())

上面這段代碼取自官方文檔，并不難讀懂，首先創(chuàng)建figure和axes，之后將涉及到的數(shù)據(jù)xdata，ydata空出來，再用update函數(shù)來更新數(shù)據(jù)并繪圖，具體為將數(shù)據(jù)賦給frames，通過FuncAnimation實例傳到update中實現(xiàn)迭代繪圖，最后可以使用ipython的內(nèi)置工具來將動態(tài)圖展示在notebook中，效果如下

到這里基本上就搞明白整個過程了，下面我們只需要照貓畫虎，寫一個繪圖函數(shù)并設(shè)計如何更新數(shù)據(jù)即可，具體如下

解釋一下這一大段代碼，拋開設(shè)置標題、坐標軸等修飾的部分，實際上也只做了一件事通過flag參數(shù)來繪制每一行數(shù)據(jù)。

為什么需分為兩段，因為我們圖上需要七根柱子，而剛開始的時候只有不到七條數(shù)據(jù)，所以需要先創(chuàng)建七個空數(shù)據(jù)，在繪制前七個柱狀圖時，依次添加，而后面只需要從7開始，刪除最早的一個數(shù)據(jù)，并向后追加一個數(shù)據(jù)即可，額外需要考慮的是我們需要為每一個數(shù)據(jù)確定一個顏色，所以還有一列顏色列也需要更新，最終我們的數(shù)據(jù)如下

在搞定shuang這個繪圖函數(shù)后，只需要迭代更新數(shù)據(jù)并展示即可，相關(guān)代碼如下

其中ani.save()可以將動圖存儲為gif，HTML(ani.to_html5_video())可以將動圖保存為mp4，而HTML(ani.to_jshtml())則是在notebook中直接查看最終效果（注意第一次運行這里會報錯，根據(jù)相關(guān)提示安裝依賴即可），大家可以自己嘗試

至于代碼中如何添加標題、修改坐標軸、修改注釋等操作本文就不在細講了，最后是制作完成的效果，我個人感覺還不錯，也可以導(dǎo)出為高分辨率視頻添加bgm制作視頻

以上就是Python利用matplotlib實現(xiàn)制作動態(tài)條形圖的詳細內(nèi)容，更多關(guān)于Python matplotlib動態(tài)條形圖的資料請關(guān)注腳本之家其它相關(guān)文章！

最新評論