欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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

程序员文章站 2022-07-02 20:04:36
Python读取指定路径所有.txt文件,将文件名写入.txt # -- coding: utf-8 -- # 生成trainval.txt文件 import os...
Python读取指定路径所有.txt文件,将文件名写入.txt
# -- 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 = './parking_data-882/training/label_kitti_format'
trainval_file = './parking_data-882/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()