參考:https://www.cnblogs.com/dengshihuang/p/8145776.html
Python獲取指定文件夾下的文件名
本文采用os.walk()和os.listdir()兩種方法，獲取指定文件夾下的文件名。
一、os.walk()
模塊os中的walk()函數(shù)可以遍歷文件夾下所有的文件。
[python] view plain copy
- os.walk(top, topdown=Ture, onerror=None, followlinks=False)
該函數(shù)可以得到一個(gè)三元tupple(dirpath, dirnames, filenames).
參數(shù)含義：
- dirpath：string，代表目錄的路徑；
- dirnames：list，包含了當(dāng)前dirpath路徑下所有的子目錄名字（不包含目錄路徑）；
- filenames：list，包含了當(dāng)前dirpath路徑下所有的非目錄子文件的名字（不包含目錄路徑）。
注意，dirnames和filenames均不包含路徑信息，如需完整路徑，可使用os.path.join(dirpath, dirnames)
下面給出代碼;
[python] view plain copy
-
-- coding: utf-8 --
import os
def file_name(file_dir):
for root, dirs, files in os.walk(file_dir):
print(root) #當(dāng)前目錄路徑
print(dirs) #當(dāng)前路徑下所有子目錄
print(files) #當(dāng)前路徑下所有非目錄子文件
當(dāng)需要特定類型的文件時(shí)，代碼如下：
[python] view plain copy
-
-- 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ù)將路徑拆分為文件名+擴(kuò)展名，例如os.path.splitext(“E:/lena.jpg”)將得到”E:/lena“+".jpg"。
二、os.listdir()
os.listdir()函數(shù)得到的是僅當(dāng)前路徑下的文件名，不包括子目錄中的文件，所有需要使用遞歸的方法得到全部文件名。
直接給出代碼，函數(shù)將返回類型為‘.jpeg’個(gè)文件名：
[python] view plain copy
-
-- 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)
