グラフ表示の際は描画、画像保存する際は消して保存したい。

Question

前提

曲線を制御点を用いて描いてますが、曲線のコードは端折ってます。

実現したいこと

グラフを表示するときは制御点を表示して、画像を保存するときは制御点を消して保存したい

該当のソースコード

# 解像度の変更
figsize_px = np.array([1024, 1024])
dpi = 100
figsize_inch = figsize_px / dpi
print("figsize", figsize_inch)  # figsize (6.4, 4.8)
fig, ax = plt.subplots(figsize=figsize_inch, dpi=dpi)

# 曲線を書く
ax.plot(px, py, color='red')

# 制御点を書く
qx = []
qy = []
for i in range(len(Q)):
    qx.append(Q[i][0])
    qy.append(Q[i][1])
ax.scatter(qx, qy, color='blue')  # , label='Control point'
ax.legend()

# メモリなんていらねえ
ax = plt.gca()
ax.axes.xaxis.set_visible(False)
ax.axes.yaxis.set_visible(False)

plt.grid(True)

# レイアウト設定
fig.tight_layout()

# グラフを表示する。
plt.show()

# ---------
n = 0
filename = fr"C:\Users\phanton\PycharmProjects\pythonProject1\kyokusen-GAN-master\kyokusen-GAN-master\kyokusenGAN\曲線データ\急カーブ\曲線{n}.jpg"

n = n + 1

# ファイルを保存する
fig.savefig(filename)

plt.close()

試したこと

いろいろ調べてやってみたのですが、私の描画の仕方と違うのでエラーが出てしまいました。
助言お願いします。

補足情報（FW/ツールのバージョンなど）

matplotlib 3.6.2

Answer 1

MatPlotLib Scatterplot removalの回答の通り、set_visible(False)あるいはremoveで消してsavefigすればとよいかと思います。

import matplotlib.pyplot as plt

fig, ax = plt.subplots()

x = y = [1,2,3]
ax.plot(x, y, color='red')
sc = ax.scatter(x, y, color='blue')
ax.legend()

# メモリなんていらねえ
ax = plt.gca()
ax.axes.xaxis.set_visible(False)
ax.axes.yaxis.set_visible(False)

plt.grid(True)

# レイアウト設定
fig.tight_layout()

# グラフを表示する。
plt.show()

sc.set_visible(False)

# ファイルを保存する
fig.savefig('ret.png')

plt.close()