matplotlib從1.1.0版本以后就開始支持繪制動畫,具體使用可以參考官方幫助文檔。下面是一個很基本的例子:
""" A simple example of an animated plot """ import numpy as np from matplotlib import pyplot as plt from matplotlib import animation # First set up the figure, the axis, and the plot element we want to animate fig = plt.figure() # create our line object which will be modified in the animation ax = plt.axes(xlim=(0, 2), ylim=(-2, 2)) # we simply plot an empty line: we’ll add data to the line later line, = ax.plot([], [], lw=2) # initialization function: plot the background of each frame def init(): line.set_data([], []) return line, # animation function. This is called sequentially # It takes a single parameter, the frame number i def animate(i): x = np.linspace(0, 2, 1000) y = np.sin(2 * np.pi * (x - 0.01 * i)) # update the data line.set_data(x, y) return line, # Makes an animation by repeatedly calling a function func # frames can be a generator, an iterable, or a number of frames. # interval draws a new frame every interval milliseconds. # blit=True means only re-draw the parts that have changed. # 在這里設置一個200幀的動畫,每幀之間間隔20毫秒 anim = animation.FuncAnimation(fig, animate, init_func=init, frames=200, interval=20, blit=True) # save the animation as an mp4. This requires ffmpeg or mencoder to be # installed. The extra_args ensure that the x264 codec is used, so that # the video can be embedded in html5. You may need to adjust this for # your system: for more information, see # http://matplotlib.sourceforge.net/api/animation_api.html anim.save('basic_animation.mp4', fps=30, extra_args=['-vcodec', 'libx264']) plt.show() # plt.show() 會一直循環播放動畫
如果要將動畫保存為mp4格式的視頻文件,則需要先安裝FFmpeg。FFmpeg是一套可以用來記錄、轉換數字音頻、視頻,並能將其轉化為流的開源計算機程序。采用LGPL或GPL許可證。它提供了錄制、轉換以及流化音視頻的完整解決方案。
在這里下載windows的版本:Download FFmpeg for Windows,解壓,然后將bin目錄加入系統環境變量的路徑中。如:C:\Program Files\ffmpeg-3.2.2-win64-static\bin。然后測試是否配置OK: 輸入ffmpeg -version
參考: