os 模塊下有兩個(gè)函數(shù): os.walk() os.listdir()
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) #當(dāng)前目錄路徑 8 print(dirs) #當(dāng)前路徑下所有子目錄 9 print(files) #當(dāng)前路徑下所有非目錄子文件 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()函數(shù)將路徑拆分為文件名+擴(kuò)展名 |
|