Python得到文件的文件名

import os
import random

# trainval_percent = 1  # trainval占总数的比例
# train_percent = 0.8   # train占trainval的比例
# xmlfilepath = r'E:\正课\大二上\计算机网络\网络编程\tensorflow-deeplab_v3_plus\data\Taidi\Annotations'
picturepath = r'E:\正课\大二上\计算机网络\网络编程\tensorflow-deeplab_v3_plus\picture'
txtsavepath = r'E:\正课\大二上\计算机网络\网络编程\tensorflow-deeplab_v3_plus\data'
# total_xml = os.listdir(xmlfilepath)
total_pic = os.listdir(picturepath)

# num = len(total_xml)
num = len(total_pic)

list = range(num)

# tv = int(num * trainval_percent)
# tr = int(tv * train_percent)
#
# trainval = random.sample(list, tv)
# train = random.sample(trainval, tr)

# ftrainval = open(txtsavepath + r'\trainval.txt', 'w')
ftest = open(txtsavepath + r'\test.txt', 'w')  # w可以覆盖
# ftrain = open(txtsavepath + r'\train.txt', 'w')
# fval = open(txtsavepath + r'\val.txt', 'w')

# for i in list:
#     name = total_pic[i][:-4] + '\n'
#     if i in trainval:
#         ftrainval.write(name)
#         if i in train:
#             ftrain.write(name)
#         else:
#             fval.write(name)


for i in list:
    name = total_pic[i][:-4] + '\n'
    ftest.write(name)

# ftrainval.close()
# ftrain.close()
# fval.close()
ftest.close()