- 溫柔點的做法
train_data_filename='../train'
if os.path.exists(train_data_filename):
#先判斷該文件夾是否已經存在,
if os.path.getsize(train_data_filename):
#如果存在,判斷該文件夾裏面是否有東西
for file in os.listdir(train_data_filename)[0:]:
#刪除該文件夾裏面的東西
#先通過os.listdir獲取該文件夾內的文件名列表,然後逐一刪除
file_path=train_data_filename+'/'+file
os.remove(file_path) #刪除該文件
else:
如果不存在,則創建該文件夾
os.mkdir(train_data_filename)
- 粗暴做法
import shutil
shutil.rmtree(validation_data_filename) #直接刪除該文件夾
os.mkdir(validation_data_filename) #創建空文件夾