一、判断文件夹是否存在
保存文件到本地时,需要判断保存路劲是否存在,如果不存在会导致保存失败。
import os
# 这里是相对路径,如果这个代码文件的同级目录下存在文件夹a且a存在文件夹b,则输出为true
print(os.path.exists('a/b')
# 这里是相对路径,如果这个代码文件的同级目录下存在文件夹a且a存在文件1.jpg,则输出为true
print(os.path.exists('a/1.jpg')
有了这两个函数,就可以判断文件或者文件夹是否存在了,然后就可以简单判断代码继续的逻辑了。
二、创建文件夹
如果文件夹存在,则直接操作,否则创建对应的文件夹之后,再继续操作
import os
if os.path.exists('a/b'):
执行写入文件操作
else:
os.mkdir('a/b')
执行写入文件操作操作
三、创建文件
如果文件存在,则打开文件,否则创建文件后打开
import os
if os.path.exists('a/1.txt'): # 如果存在a/1.txt文件
r = os.path.join("a", str(1)+".txt")
file = open(r, 'a') # a表示在原有文档下继续编辑
file.write('\n' + 'loss: ' + str(2) + '\n') # 在文件中写入loss: 2
file.flush()
file.close()
else:
# r为路径,'a\\1.txt','\\'与'/'一致,在os.path.join()函数中,
# 每个逗号之间的元素都会被'\\'分开,无需添加分隔符
r = os.path.join("a", str(1)+".txt")
file = open(r, 'w') # w表示在重新创建文件
file.write('\n' + 'loss: ' + str(2) + '\n') # 在文件中写入loss: 2
file.flush()
file.close()