首先給出代碼：

import os    path = "/home/data//"path_exp = os.path.expanduser(path)classes = [int(p) for p in os.listdir(path_exp)]classes.sort()# nrof_classes一個數據集下有多少個文件夾,就是說有多少個人,多少個類別nrof_classes = len(classes)count=0files = open("train_list.txt",'w')filets = open("test_list.txt",'w')count_u=0for i in range(nrof_classes):  class_name = str(classes[i])  count=count+1  count_u=count_u+1  facedir = os.path.join(path_exp, class_name)  prefix1 = path+class_name+"/"  if os.path.isdir(facedir):    images = os.listdir(facedir)    #print(images[0])    image_paths = [(prefix1+img+" "+class_name+"/n") for img in images]    #print(image_paths[0])    if count < 0.8*nrof_classes:      if len(image_paths)>4:        test_path=[]        for x in range(2):          test_path.append(image_paths[0])          del image_paths[0]        filets.writelines(test_path)    files.writelines(image_paths)    #if count==2:    #  break    #imgae_pathses = []    #防止圖像大小為0    #for x in image_paths:    #  if os.path.getsize(x)>0:    #    imgae_pathses.append(x)    #if len(imgae_pathses)==0:    #  os.rmdir(facedir)files.close()filets.close()

python下os模塊的一下有用的用法：

0 重命名：文件和文件夾都是一個命令：

os.rename(original_dir,new_dir)

1 文件操作：

os.mknod("test.txt")        創建空文件
fp = open("test.txt",w)     直接打開一個文件，如果文件不存在則創建文件

關于open 模式：

w     以寫方式打開，
a     以追加模式打開 (從 EOF 開始, 必要時創建新文件)
r+     以讀寫模式打開
w+     以讀寫模式打開 (參見 w )
a+     以讀寫模式打開 (參見 a )
rb     以二進制讀模式打開
wb     以二進制寫模式打開 (參見 w )
ab     以二進制追加模式打開 (參見 a )
rb+    以二進制讀寫模式打開 (參見 r+ )
wb+    以二進制讀寫模式打開 (參見 w+ )
ab+    以二進制讀寫模式打開 (參見 a+ )

fp.read([size])                     #size為讀取的長度，以byte為單位

fp.readline([size])                 #讀一行，如果定義了size，有可能返回的只是一行的一部分

fp.readlines([size])                #把文件每一行作為一個list的一個成員，并返回這個list。其實它的內部是通過循環調用readline()來實現的。如果提供size參數，size是表示讀取內容的總長，也就是說可能只讀到文件的一部分。