<del id="uwgwi"></del>

極簡教程: 使用 matplotlib 繪制 GIF 動圖

fredshare 發(fā)布于2019-07-31 10:04 / 2596人閱讀

摘要：如果你保存成的時候，你需要提供繪制一個散點圖不會重繪，和初始的線更新線和坐標(biāo)軸標(biāo)簽返回要重繪的對象將會在每一幀執(zhí)行一次幀數(shù)每幀的間隔如果第一參數(shù)是，教會保存成重點保存的尺寸使用的渲染器，我們制定成否則直接展示

開門見山，直接上例子：

有如下特點：

散點圖的部分是不變的；線是移動的

X 軸標(biāo)題每一禎改變一次

DEMO 的環(huán)境

Ubuntu 18.04.2 LTS

conda 4.6.3

Python 3.7.2

創(chuàng)建 virtualenv

ichexw at n3xt-Studio -> conda create --name matplot-gif python=3.7
ichexw at n3xt-Studio -> conda activate matplot-gif

安裝必要的依賴

安裝 matplotlib

(matplotlib-gif) ichexw at n3xt-Studio -> conda install matplotlib

安裝 imagemagick

(matplotlib-gif) ichexw at n3xt-Studio -> conda install -c conda-forge imagemagick

代碼實現(xiàn)

import sys
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

# 創(chuàng)建圖層和布局
fig, ax = plt.subplots()
fig.set_tight_layout(True)

# 查看圖標(biāo)的尺寸。如果你保存成 gif 的時候，你需要提供 DPI
print("fig size: {0} DPI, size in inches {1}".format(
    fig.get_dpi(), fig.get_size_inches()))

# 繪制一個散點圖（不會重繪），和初始的線
x = np.arange(0, 20, 0.1)
ax.scatter(x, x + np.random.normal(0, 3.0, len(x)))
line, = ax.plot(x, x - 5, "r-", linewidth=2)

def update(i):
    label = "timestep {0}".format(i)
    print(label)
    
    # 更新線和坐標(biāo)軸標(biāo)簽
    line.set_ydata(x - 5 + i)
    ax.set_xlabel(label)
    
    # 返回要重繪的對象
    return line, ax

if __name__ == "__main__":
    # FunAnimation 將會在每一幀執(zhí)行一次 update
    # frames: 幀數(shù)
    # interval: 每幀的間隔
    anim = FuncAnimation(fig, update, frames=np.arange(0, 10), interval=200)
    if len(sys.argv) > 1 and sys.argv[1] == "save":
        # 如果第一參數(shù)是 save，教會保存成 gif
        # **重點**
        # dpi: 保存的尺寸
        # writer: 使用的渲染器，我們制定成 imagemagick
        anim.save("line.gif", dpi=80, writer="imagemagick")
    else:
        # 否則直接展示
        plt.show()