python 固定文件名字长度并保存

文件夹下面的图片名字格式为000001.jpg,001250.jpg。。。
import os

ROOT_DIR = os.path.abspath("./")
train_DIR = os.path.join(ROOT_DIR, "darknet/VOCdevkit/VOC2018/JPEGImages/train")

dest = os.path.join(ROOT_DIR, 'darknet/VOCdevkit/VOC2018/ImageSets/Main/train.txt')
file_list = os.listdir(train_DIR)

train_file = open(dest, 'w')
a = []
for file_obj in file_list:
    file_name, file_extend = os.path.splitext(file_obj)＃读取文件名字，并分割000001和.jpg
    int_filename = int(file_name)＃将str类型的000001转为int型
    a.append(int_filename)
a = sorted(a)＃从小到大排序，此时的a为[1，1250]
print(a)

for file_obj in a:
    file_obj = "%06d" % file_obj#固定长度输出[000001,001250]
    train_file.write(str(file_obj) + '\n')#int型转化为str保存到txt文件
train_file.close()