通过创建新文件夹保存图形

Question

我目前正在通过手动创建一个文件夹来保存我的地块。但我想自动创建文件夹并将图形保存在其中。我当前的代码如下所示

plt.savefig('Figures_Voltage/Pack voltage.png')

Figures_Voltage 是我手动创建的文件夹，里面有 .py 文件，但我希望该文件夹由 Python 创建。谁能帮帮我。提前致谢

Answer 1

您可以检查文件夹是否不存在并创建它

import os

if not os.path.exists('Figures_Voltage'):
    os.mkdir('Figures_Voltage')

如果您在它已经存在时尝试创建，则会引发错误。

或者如果文件夹已经存在则使用 makedirs 和 exist_ok=True 跳过它（不会引发错误）

os.makedirs('Figures_Voltage', exist_ok=True)

makedirs 也可以一次创建 folder/subfolder/subsubfolder。

Saving figure by creating new folder