Python保存图片的三种情况

1、保存单张图片

#基本存储
import matplotlib.pyplot as plt
x=[0,1,2,3,4,5]
y=[0,2,4,6,8,10]
plt.plot(x,y)#绘制图片
plt.savefig( './result/exam_01.png')  #将图片存储在result文件夹下并命名为exam_01.png  ，注意该行代码要放在plt.show()前
plt.show()

2、创建文件夹并保存单张图片

#存储图片同一个文件夹中
import matplotlib.pyplot as plt
import os#导入OS模块，用于创建文件夹
x=[0,1,2,3,4,5]
y=[0,2,4,6,8,10]
plt.plot(x,y)
# 指定图片保存路径
figure_save_path = "picture_folder"

if not os.path.exists(figure_save_path):
    os.makedirs(figure_save_path) # 如果不存在目录figure_save_path，则创建
    
plt.savefig(os.path.join(figure_save_path , 'exam_02.png'))#第一个是指存储路径，第二个是图片名字
plt.show()

3、创建文件夹并保存多张图片

#批量绘制图片+批量存储
import matplotlib.pyplot as plt
import os
for i in range(0,3,1):
    x=[i,1,2,3,4,5]
    y=[0,2,4,6,8,10]
    name_list=["one","two","three"]#先创建名字列表，可以是用在图片命名，也可以用在文件夹命名
    plt.plot(x,y)
    # 指定图片保存路径
    figure_save_path = "picture_folder_many"#这里创建了一个文件夹，如果依次创建不同文件夹，可以用name_list[i]
    if not os.path.exists(figure_save_path):
        os.makedirs(figure_save_path) # 如果不存在目录figure_save_path，则创建
    plt.savefig(os.path.join(figure_save_path , name_list[i]))#分别命名图片
    plt.show()