Python獲取指定文件夾下的文件名

本文采用os.walk()和os.listdir()兩種方法，獲取指定文件夾下的文件名。

 

一、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)

 

下面給出代碼;

 

# -*- 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()函數將路徑拆分為文件名+擴展名，例如os.path.splitext(“E:/lena.jpg”)將得到”E:/lena“+".jpg"。

 

 

二、os.listdir()

 

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

 

直接給出代碼，函數將返回類型為‘.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解釋器，否則仍將運行第一次執行過的代碼。