本文采用os.walk()和os.listdir()兩種方法,獲取指定文件夾下的文件名。
一、os.walk()
模塊os中的walk()函數(shù)可以遍歷文件夾下所有的文件。
os.walk(top, topdown=Ture, onerror=None, followlinks=False)
該函數(shù)可以得到一個三元tupple(dirpath, dirnames, filenames).
參數(shù)含義:
注意,dirnames和filenames均不包含路徑信息,如需完整路徑,可使用os.path.join(dirpath, dirnames)
下面給出代碼;
# -*- coding: utf-8 -*- import os def file_name(file_dir): for root, dirs, files in os.walk(file_dir): print(root) #當前目錄路徑 print(dirs) #當前路徑下所有子目錄 print(files) #當前路徑下所有非目錄子文件
當需要特定類型的文件時,代碼如下:
# -*- coding: utf-8 -*- import os def file_name(file_dir): L=[] for root, dirs, files in os.walk(file_dir): for file in files: if os.path.splitext(file)[1] == '.jpeg': L.append(os.path.join(root, file)) return L
其中os.path.splitext()函數(shù)將路徑拆分為文件名+擴展名,例如os.path.splitext(“E:/lena.jpg”)將得到”E:/lena“+".jpg"。
二、os.listdir()
os.listdir()函數(shù)得到的是僅當前路徑下的文件名,不包括子目錄中的文件,所有需要使用遞歸的方法得到全部文件名。
直接給出代碼,函數(shù)將返回類型為‘.jpeg'個文件名:
# -*- coding: utf-8 -*- import os def listdir(path, list_name): for file in os.listdir(path): file_path = os.path.join(path, file) if os.path.isdir(file_path): listdir(file_path, list_name) elif os.path.splitext(file_path)[1]=='.jpeg': list_name.append(file_path)
初學python,犯了些低級錯誤,浪費了很多時間,首先一定要確保代碼的四空格縮進,注意直接回車下一行并不是四空格縮進;還有一點是在命令提示符窗口(Windows)使用python解釋器第一次運行代碼后,修改源代碼,如若再次運行需要退出重新進入python解釋器,否則仍將運行第一次執(zhí)行過的代碼。
以上就是本文的全部內(nèi)容,希望對大家的學習有所幫助,也希望大家多多支持武林站長站。
新聞熱點
疑難解答