pyplotで生成した散布図などの保存方法について

Question

import numpy as np
from matplotlib import pyplot as plt

x = np.zeros(30)
for n in range(30):
    x[n] = n
plt.scatter( x, x )
plt.show()
# im = ?
# plt.imsave("image.jpg", im)

このような画像を保存したいのですが、どのようにすればよいですか？

Answer 1

showする前にsavefigすればOKです。

import numpy as np
from matplotlib import pyplot as plt

x = np.zeros(30)
for n in range(30):
    x[n] = n
plt.scatter( x, x )
plt.savefig("image.png") # showする前に書き出す。showしたら消えてしまう
plt.show()