Python 動くグラフを作ってみた!（5分）

#数値計算を行うためのライブラリ
import numpy as np

#グラフを作成するためのライブラリ
import matplotlib.pyplot as plt

#グラフのアニメーションを作成するためのライブラリ
import matplotlib.animation as animation

#upyter上でのインタラクティブな画像表示を実現するための機能
%matplotlib nbagg

#グラフを用意
fig = plt.figure()

#タイトルを追加
plt.title("SIN&COS")

#xラベルを追加
plt.xlabel("x")

#yラベルを追加
plt.ylabel("y")

#グラフの表示範囲を決める
x = np.arange(0, 10, 0.1)

#リストを用意
ims = []

#100回繰り返す
for a in range(100):
    
    #サイン
    y_sin = np.sin(x - a)
    
    line_sin, = plt.plot(x, y_sin, color="r")
    

    #コサイン
    y_cos = np.cos(x - a)
    
    line_cos, = plt.plot(x, y_cos, color="b")
    

    #リストに追加していく
    ims.append([line_sin, line_cos])

#動的なグラフを作成する
ani = animation.ArtistAnimation(fig, ims)

#凡例を追加（loc="upper right"→右上に表示させる）
plt.legend(("sin", "cos"), loc="upper right")

#グラフを用意
fig = plt.figure()

#タイトルを追加
plt.title("Trigonometric function")

#xラベルを追加
plt.xlabel("x")

#yラベルを追加
plt.ylabel("y")

#グラフの表示範囲を決める
x2 = np.arange(0, 10, 0.1)

#リストを用意
ims2 = []

for a in range(100):
    
    #サイン
    y2 = np.sin(x2 - a) + np.cos(2*(x2 - a))
    
    line2, = plt.plot(x2, y2, color="g")
        
    #リストに追加していく
    ims2.append([line2])

#動的なグラフを作成する
ani2 = animation.ArtistAnimation(fig, ims2)

plt.legend(["sinx+cos2x"], loc="upper right")