matplotlib.animation を使わずにアニメーションを保存する方法

Question

matplotlib.animation を使用せず、matplotlib.pyplot で描画しては消して、という方法でアニメーションを作成しました。
（実際のコードを見ていただいた方が意図を理解しやすいと思います。）

出力結果を動画として保存するにはどのようにすれば良いでしょうか。
そのような方法がありましたら、教えていただきたいです。

#!/usr/bin/env python3

import matplotlib.pyplot as plt
import numpy as np

show_animation = True


def update(x):
    x[0] += 0.1
    x[1] += 0.0
    return x


def main():
    print(__file__ + 'Start!')
    
    # initialization
    p_start = np.array([0.0, 2.0])
    p_goal = np.array([10.0, 2.0])

    x = np.array([0.0, 2.0])

    while True:
        # update
        x = update(x)

        if show_animation:
            plt.cla()

            plt.gcf().gca().set_aspect('equal')
            plt.grid(True)
            plt.xlim(-0.5, 10.5)
            plt.ylim(-0.5, 4.5)

            plt.plot(p_start[0], p_start[1], '.r', markersize=12)
            plt.plot(p_goal[0], p_goal[1], '.b', markersize=12)

            plt.plot(x[0], x[1], '^g')

            plt.pause(0.001)

        if x[0] >= 10:
            print('Goal!')
            break

    plt.show()


if __name__ == '__main__':
    main()

Answer 1

imageio を使う方法です。

#!/usr/bin/env python3

import matplotlib.pyplot as plt
import numpy as np
import imageio
import io

show_animation = True

def update(x):
    x[0] += 0.1
    x[1] += 0.0
    return x

def main():
    print(__file__ + 'Start!')
    
    # initialization
    p_start = np.array([0.0, 2.0])
    p_goal = np.array([10.0, 2.0])

    x = np.array([0.0, 2.0])
    anim = imageio.get_writer('anim.gif', mode='I', duration=0.1)

    while True:
        # update
        x = update(x)

        if show_animation:
            plt.cla()

            plt.gcf().gca().set_aspect('equal')
            plt.grid(True)
            plt.xlim(-0.5, 10.5)
            plt.ylim(-0.5, 4.5)

            plt.plot(p_start[0], p_start[1], '.r', markersize=12)
            plt.plot(p_goal[0], p_goal[1], '.b', markersize=12)

            plt.plot(x[0], x[1], '^g')

            # append plot data to animation gif file 
            plt.gcf().canvas.draw()
            img = np.frombuffer(plt.gcf().canvas.tostring_rgb(), dtype=np.uint8)
            img = img.reshape(plt.gcf().canvas.get_width_height()[::-1] + (3,))
            anim.append_data(img)

            plt.pause(0.001)

        if x[0] >= 10:
            print('Goal!')
            break

    plt.show()
    anim.close()

if __name__ == '__main__':
    main()

Comments

[Ryuking885]

非常に助かりました。ありがとうございました！