1. 程式人生 > >利用python.os.listdir獲取資料夾中所有檔名

利用python.os.listdir獲取資料夾中所有檔名

利用內建os的listdir可以獲取資料夾中的檔名,從而可以利用檔名對資料做進一步的處理,比如:在貓狗識別的練習過程中,需要通過檔名對貓狗資料進行分類。

def get_file(file_dir):
    dogs = []
    cats = []
    dog_labels = []
    cat_labels = []
    #利用listdir可以獲取資料夾中所有檔案的名稱
    for dir in os.listdir(file_dir):
        temp = dir.split(".")
        if temp[0] == 'dog':
            dogs.append(file_dir+dir)
            dog_labels.append(1
) else: cats.append(file_dir+dir) cat_labels.append(0)