如何将一系列matplotlib图保存为图像文件？

import matplotlib.pyplot as plt
import os

output_folder = 'output'
os.makedirs(output_folder, exist_ok=True)

# 生成图像1
plt.plot([1, 2, 3, 4])
plt.savefig(os.path.join(output_folder, 'image1.png'))
plt.close()

# 生成图像2
plt.scatter([1, 2, 3, 4], [5, 6, 7, 8])
plt.savefig(os.path.join(output_folder, 'image2.png'))
plt.close()

# 生成图像3
plt.bar(['A', 'B', 'C', 'D'], [3, 5, 2, 7])
plt.savefig(os.path.join(output_folder, 'image3.png'))
plt.close()