Python读取指定路径所有.txt文件，将文件名写入.txt

最新推荐文章于 2024-09-10 22:25:23 发布

菜鸡信息技术

最新推荐文章于 2024-09-10 22:25:23 发布

阅读量7.2k

点赞数 2

分类专栏： Program 文章标签： python

本文链接：https://blog.csdn.net/u011440558/article/details/77885036

版权

Program 专栏收录该内容

19 篇文章 1 订阅

订阅专栏

# -- coding: utf-8 --
# 生成trainval.txt文件
import os  

def file_name(file_dir):   
    L=[]   
    for root, dirs, files in os.walk(file_dir):  
        for file in files:  
            if os.path.splitext(file)[1] == '.txt':  
                # L.append(os.path.join(root, file))  
                file_name = file[0:-4]  #去掉.txt
                L.append(file_name)  
    return L  
    
label_folder = './training/label_kitti_format'
trainval_file = './ImageSets/trainval.txt'

txt_name = file_name(label_folder)

with open(trainval_file, 'w') as f:
  for i in txt_name:
    f.write('{}\n'.format(i))
f.close()