Python 读取文件夹名字(不包括后缀)并保存为txt文件

 将train文件夹下的文件保存到train.txt中,代码如下所示:

import os

path = "D:/Code/Data/roaddata/MassachusettsRoadsDataset/tiff/val"
file_name_list = os.listdir(path)

with open('D:/Code/Data/roaddata/MassachusettsRoadsDataset/list/valid.txt','w') as f:
    for file_name in file_name_list:
        f.write(file_name[0:-5]+'\n') # -5表示从后往前数,到小数点位置

 过程及结果如下:

Python 读取文件夹名字(不包括后缀)并保存为txt文件_第1张图片

Python 读取文件夹名字(不包括后缀)并保存为txt文件_第2张图片

 

你可能感兴趣的:(python)