python创建train list和test list

python写循环从000000到059999：

for i in range(60000):
     '{0:06}'.format(i)

读某个文件夹并将文件名写入txt文件：

import os
mydir = r'/home/test/'
 
if __name__ == '__main__':
    for pathstr,dirlist,filelist in os.walk(mydir):
        for file in filelist:
            with open("file.txt","ab") as f:
                f.write(file+'\n')

创建train list和test list

for i in range(48000)：
    num = '{0:06}'.format(i)
    with open("train_data_list.txt","ab") as f: 
        f.write(num + '\n')    

for i in range(48001，60000)：
    num = '{0:06}'.format(i)
    with open("test_data_list.txt","ab") as f: 
        f.write(num + '\n')