Python 獲取文件夾下的文件名

os 模塊下有兩個函數：

1. os.walk()
2. os.listdir()

1、os.walk()

模塊os中的walk()函數可以遍歷文件夾下所有的文件。

os.walk(top, topdown=Ture, onerror=None, followlinks=False)  

該函數可以得到一個三元元組 tupple(dirpath, dirnames, filenames).

參數含義：

• dirpath：string，代表目錄的路徑；
• dirnames：list，包含了當前dirpath路徑下所有的子目錄名字（不包含目錄路徑）；
• filenames：list，包含了當前dirpath路徑下所有的非目錄子文件的名字（不包含目錄路徑）。

注意，dirnames和filenames均不包含路徑信息，如需完整路徑，可使用os.path.join(dirpath, dirnames)

下面給出代碼;

#  !/usr/bin/env  python
#  -*- coding:utf-8 -*-
# @Time   :  2019.
# @Author :  綠色羽毛
# @Email  :  [email protected]
# @Blog   :  https://blog.csdn.net/ViatorSun
# @Note   :  


import os  
  
def file_name(file_dir):   
    for root, dirs, files in os.walk(file_dir):  
        print(root) #當前目錄路徑  
        print(dirs) #當前路徑下所有子目錄  
        print(files) #當前路徑下所有非目錄子文件  

當需要特定類型的文件時，代碼如下：

#  !/usr/bin/env  python
#  -*- coding:utf-8 -*-
# @Time   :  2019.
# @Author :  綠色羽毛
# @Email  :  [email protected]
# @Blog   :  https://blog.csdn.net/ViatorSun
# @Note   :  
  
  
import os  
  
def file_name(file_dir):   
    FileList=[]   
    for root, dirs, files in os.walk(file_dir):  
        for file in files:  
            if os.path.splitext(file)[1] == '.jpg':  
                FileList.append(os.path.join(root, file))  
    return FileList  

其中os.path.splitext()函數將路徑拆分爲文件名+擴展名，例如os.path.splitext(“E:\IDCard\love.jpg”)將得到 “E:\IDCard\love”+".jpg"。

2、os.listdir()

os.listdir()函數得到的是指定路徑下的文件名，不包括子目錄中的文件，所有需要使用遞歸的方法得到全部文件名。

直接給出代碼，函數將返回類型爲‘.jpg’個文件名：

#  !/usr/bin/env  python
#  -*- coding:utf-8 -*-
# @Time   :  2019.
# @Author :  綠色羽毛
# @Email  :  [email protected]
# @Blog   :  https://blog.csdn.net/ViatorSun
# @Note   :  


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]=='.jpg':  
            list_name.append(file_path)