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