python - 将 matplotlib 文件保存到目录

Question

这是一个简单的代码，它生成一个绘图图像并将其保存在与代码相同的目录中。现在，有没有办法可以将它保存在选择的目录中？

import matplotlib
import matplotlib.pyplot as plt

fig = plt.figure()
ax = fig.add_subplot(111)
ax.plot(range(100))

fig.savefig('graph.png')

Answer 1

如果您要保存的目录是工作目录的子目录，只需在文件名前指定相对路径即可:

    fig.savefig('Sub Directory/graph.png')

如果您希望使用绝对路径，请导入 os模块:

    import os
    my_path = os.path.abspath(__file__) # Figures out the absolute path for you in case your working directory moves around.
    ...
    fig.savefig(my_path + '/Sub Directory/graph.png')

如果不想担心子目录名前面的斜杠，可以如下智能加入路径:

    import os
    my_path = os.path.abspath(__file__) # Figures out the absolute path for you in case your working directory moves around.
    my_file = 'graph.png'
    ...
    fig.savefig(os.path.join(my_path, my_file))