python 获取一个文件夹里面全部文件名（不加后缀），并写到txt里面

时间 2020-07-25 标签 python 获取一个文件夹里面全部文件名不加后缀写到 txt 里面

import os
names = os.listdir('E:/dir')  #路径
i=0  #用于统计文件数量是否正确，不会写到文件里
train_val = open('test.txt','w')
for name in names:
    index = name.rfind('.')
    name = name[:index]
    train_val.write(name+'\n')
    i=i+1
print i