matplotlib animation FuncAnimation畫2D線圖

覺得有用的話,歡迎一起討論相互學習~

效果圖

在這里插入圖片描述

代碼

import numpy as np
from matplotlib import pyplot as plt
from matplotlib import animation

fig, ax = plt.subplots()
# 生成子圖，相當於fig = plt.figure(),
# ax = fig.add_subplot(),其中ax的函數參數表示把當前畫布進行分割，
# 例：fig.add_subplot(2,2,2).表示將畫布分割為兩行兩列，ax在第2個子圖中繪制，其中行優先。
x = np.arange(0, 2*np.pi, 0.01)  # 表示從0~2*np.pi之間每隔0.01取一個點
line, = ax.plot(x, np.sin(x))  # 注意，這里line后面要加上逗號，表示一個具有一個元素的元組


# print(type(line))
# print(type((line,)))
# <class 'matplotlib.lines.Line2D'>
# <class 'tuple'>

def animate(i):  # 這里的i其實就是參數0-99，即時frames控制的參數，控制程序畫圖變換的次數
    # print(i)  # 0-99
    line.set_ydata(np.sin(x + i/10.0))  # 改變線條y的坐標值
    return line,


def init():  # 初始化函數，圖形開始顯示的狀態
    line.set_ydata(np.sin(x))
    return line,


ani = animation.FuncAnimation(fig=fig, func=animate, frames=100, init_func=init,interval=20, blit=False)
"""frames設定幀數,總共執行100個update就會進行下一次循環，並且frames還會作為參數傳入animate()函數，init_func設定初始函數圖像,
interval設置更新間隔此處設置為20毫秒，(仔細想想20毫秒其實是很小的一個間隔)
blit如果是只有變化了的像素點才更新就設置為True,如果是整張圖片所有像素點全部更新的話就設置為False
"""
plt.show()