os 模塊下有兩個函數:app
os.walk()函數
os.listdir()spa
1 # -*- coding: utf-8 -*- 2 3 import os 4 5 def file_name(file_dir): 6 for root, dirs, files in os.walk(file_dir): 7 print(root) #當前目錄路徑 8 print(dirs) #當前路徑下全部子目錄 9 print(files) #當前路徑下全部非目錄子文件
1 # -*- coding: utf-8 -*- 2 3 import os 4 5 def file_name(file_dir): 6 L=[] 7 for root, dirs, files in os.walk(file_dir): 8 for file in files: 9 if os.path.splitext(file)[1] == '.jpeg': 10 L.append(os.path.join(root, file)) 11 return L 12 13 14 #其中os.path.splitext()函數將路徑拆分爲文件名+擴展名
1 # -*- coding: utf-8 -*- 2 import os 3 4 def listdir(path, list_name): #傳入存儲的list 5 for file in os.listdir(path): 6 file_path = os.path.join(path, file) 7 if os.path.isdir(file_path): 8 listdir(file_path, list_name) 9 else: 10 list_name.append(file_path)