Python如何列出目錄下所有文件

一、獲取當前工作目錄

程序在運行時，需要知道當前所在的工作目錄，才能夠列出該目錄下的文件。Python提供了os模塊來處理文件和目錄的操作。os.getcwd()函數可以返回當前工作目錄並存儲到一個變量中。

import os

# 獲取當前工作目錄
cwd = os.getcwd()
print(cwd)  # 輸出：/User/test/

上述代碼中，我們使用os模塊的getcwd()函數獲取當前工作目錄，並將其保存到變量cwd中。輸出結果為當前工作目錄的路徑。

二、列出目錄下的所有文件

在獲取當前工作目錄之後，我們需要使用os.listdir()函數列出該目錄下的所有文件。該函數返回一個列表，其中包含文件夾和文件的名稱。我們可以使用for循環遍歷該列表，並使用os.path.join()函數將路徑和文件名結合起來，以獲取每個文件的完整路徑。

import os

# 獲取當前工作目錄
cwd = os.getcwd()

# 列出目錄下的所有文件
for filename in os.listdir(cwd):
    # 獲取文件的完整路徑
    file_path = os.path.join(cwd, filename)
    # 判斷是否是文件
    if os.path.isfile(file_path):
        print(file_path) # 輸出文件的完整路徑
    elif os.path.isdir(file_path):
        print("文件夾：" + file_path) # 輸出文件夾的路徑

上述代碼中，我們遍歷了當前工作目錄下的所有文件和文件夾，並使用os.path.join()函數將路徑和文件名組合起來，以獲得每個文件和文件夾的完整路徑。我們使用os.path.isfile()函數判斷該路徑是否為文件，並輸出完整路徑。如果該路徑為文件夾，則使用os.path.isdir()函數判斷，並輸出文件夾的路徑。

三、遞歸地列出目錄下的所有文件

在需要列出所有文件的子目錄時，我們需要使用遞歸函數。一個遞歸函數是在函數內部調用自己，直到某個條件滿足為止。在列出目錄下的所有文件時，我們需要遍歷當前目錄下的所有文件和文件夾，並查看每個文件夾是否包含子目錄，如果有，則繼續遍歷子目錄中的文件。這裡我們使用os.walk()函數，該函數遞歸地遍歷目錄結構，並返回包含當前目錄中所有文件和文件夾名稱的元組。

import os

def list_files(path):
    # 遞歸地列出目錄下的所有文件
    for root, dirs, files in os.walk(path):
        for filename in files:
            # 獲取文件的完整路徑
            file_path = os.path.join(root, filename)
            print(file_path) # 輸出文件的完整路徑

# 獲取當前工作目錄
cwd = os.getcwd()

# 列出目錄下的所有文件
list_files(cwd)

上述代碼中，我們定義了一個名為list_files()的函數，該函數使用os.walk()函數遞歸遍歷目錄結構並列出指定路徑下所有的文件路徑。我們在最後調用該函數，並傳入當前工作目錄作為參數。

四、根據文件名後綴篩選文件

在實際應用中，我們可能需要只列出特定類型的文件。在Python中，我們可以使用字符串方法來實現這一操作。我們首先使用os.path.splitext()函數來分離文件路徑和文件名後綴。然後，我們可以使用字符串方法來檢查文件名後綴是否符合要求。

import os

def list_files_by_extension(path, extension):
    # 遞歸地列出目錄下指定後綴名的所有文件
    for root, dirs, files in os.walk(path):
        for filename in files:
            # 獲取文件的完整路徑
            file_path = os.path.join(root, filename)
            # 分離文件名和文件名後綴，獲取文件名後綴
            _, file_extension = os.path.splitext(file_path)
            # 判斷文件名後綴是否匹配
            if file_extension == extension:
                print(file_path) # 輸出文件的完整路徑

# 獲取當前工作目錄
cwd = os.getcwd()

# 列出目錄下的所有.png文件
list_files_by_extension(cwd, ".png")

上述代碼中，我們定義了一個名為list_files_by_extension()的函數，該函數可以篩選出指定的文件類型。我們首先使用os.path.splitext()函數獲取文件的後綴名，然後判斷該後綴名是否匹配我們想要的後綴名，並輸出完整文件路徑。

五、結論

在本文中，我們深入探討了如何使用Python列出目錄下的所有文件。從獲取當前工作目錄開始，我們通過os模塊提供的函數和方法，介紹了多個列出文件的方法。此外，我們還介紹了如何遞歸地列出目錄下的所有文件，以及如何根據文件名後綴來篩選指定類型的文件。這些方法將在文件操作中提供幫助。