我一直在寻找一种简单的方法来在我的脚本开头设置一个工作文件夹,以避免每次从另一台机器运行我的脚本时都要更改它。对不起,如果这是愚蠢的,但我无法深入了解奇怪功能的海洋,我无法找到答案 - 而且我不知道如何游泳。
我会用它将所有图像保存到同一个特定文件夹中。
我的看法:
#import stuff
filepath=raw_input('Provide the working directory: ')
#do stuff
plt.savefig(filepath+'\\image.jpg', dpi=2000,bbox='tight')
修改
我有一些文件要保存在一个特定的目录中,与我的脚本所在的位置相同。但它是这样指定的:C:\\Users\\User\\Desktop\\N_Scripts
它在我自己的电脑上,但我经常在不同的机器上工作。如何使用raw_input
之类的内容预设脚本开头所有图像所在的目录?
感谢您的帮助!
答案 0 :(得分:1)
保存在一个特定目录中,与我的脚本所在的目录相同
将Python脚本中的当前工作目录更改为脚本所在的目录:
import os
os.chdir(get_script_dir())
其中get_script_dir()
is defined here。
但是,一旦你使用os.chdir,我如何修改plt.savefig的参数以确保我的图像保存在我的脚本所在的位置? 它只是plt.savefig(os.chdir(get_script_dir())+' \ image.jpg)?
使用plt.savefig('image.jpg')
。