matplotlibを使用した散布図アニメーションの保存

matplotlibを使用して散布図アニメーションを保存するには、次の手順を実行できます-

• 図のサイズを設定し、サブプロット間およびサブプロットの周囲のパディングを調整します。
• 4つの変数、ステップ、ノード、位置、およびソリューションを初期化します。
• リストに位置とソリューションの値を追加します。
• 図とサブプロットのセットを作成します。
• マーカーサイズの変数を初期化します。
• グリッド線を構成します。
• 関数を繰り返し呼び出してアニメーションを作成します*アニメーション *、軸をクリアするには、新しい軸サブロットを追加し、軸に散布点をプロットします。
• アニメーション化された散布図を.gifとして保存します ファイル。

例

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

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

steps = 50
nodes = 100
positions = []
solutions = []

for i in range(steps):
   positions.append(np.random.rand(2, nodes))
   solutions.append(np.random.random(nodes))

fig, ax = plt.subplots()
   marker_size = 50

def animate(i):
   fig.clear()
   ax = fig.add_subplot(111, aspect='equal', autoscale_on=False, xlim=(0, 1), ylim=(0, 1))
   ax.set_xlim(0, 1)
   ax.set_ylim(0, 1)
   s = ax.scatter(positions[i][0], positions[i][1], s=marker_size, c=solutions[i], cmap="RdBu_r", marker="o", edgecolor='black')

plt.grid(b=None)
ani = animation.FuncAnimation(fig, animate, interval=100, frames=range(steps))

ani.save('animation.gif', writer='pillow')

出力