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()