一、OS模塊的介紹
在Python中,操作系統(OS)模塊提供了一種與操作系統進行交互的編程方式,它能夠讓你調用操作系統的底層函數和系統調用,這些函數和調用一般由C編寫並且需要使用系統內核提供的API調用。
OS模塊在Python中常用於管理文件和目錄以及執行其他與操作系統相關的任務。這個模塊的主要目的是讓Python程序能夠執行基本的系統操作,比如創建一個文件夾,列出目錄中的文件,刪除一個文件等。
二、使用OS模塊讀取目錄文件列表
使用OS模塊可以輕鬆讀取一個目錄中的文件列表。可以通過以下代碼來讀取指定目錄的文件列表:
import os # 獲取指定目錄的全部文件列表 def get_file_list(path): files = os.listdir(path) return files # 調用函數獲取當前目錄的文件列表 path = "." files = get_file_list(path) print(files)
代碼執行結果如下所示:
['file1.txt', 'file2.png', 'subdir']
上面的代碼首先導入了Python中的OS模塊,並定義了函數get_file_list
,這個函數接收一個參數path,返回一個包含當前目錄下所有文件的列表。
在函數中,調用了OS模塊中的listdir
函數,這個函數會返回指定目錄中的所有文件和子目錄的列表。在示例中,我們將.
傳遞給函數,表示當前目錄。在實際使用中,可以將一個路徑字符串傳遞給函數以獲取該路徑下的文件列表。
最後,我們在main
函數中調用get_file_list
函數來獲取當前目錄下的文件列表,並打印輸出。
三、處理子目錄
在實際使用中,也可以使用遞歸方法,來獲取目錄樹下的所有文件列表,包括子目錄中的文件。下面的示例代碼演示了如何讀取一個目錄及其子目錄下所有文件的列表。
import os # 遞歸函數獲取指定目錄的全部文件列表 def get_all_file_list(path): result = [] for root, dirs, files in os.walk(path): for file in files: result.append(os.path.join(root, file)) return result # 調用函數獲取文件列表 path = "." files = get_all_file_list(path) print(files)
代碼執行結果如下所示:
['./file1.txt', './file2.png', './subdir/file3.txt']
在上面的代碼中,我們定義了一個遞歸函數get_all_file_list
,並使用os.walk
函數逐層遍歷指定目錄的所有子目錄,以獲取所有目錄下的文件列表。
對於每個訪問到的目錄,os.walk函數都會返回一個包含三個元素的元組:根路徑、目錄列表和文件列表。我們只需要將每個文件的完整路徑添加到結果列表中即可。通過使用os.path.join
函數,可以將目錄名和文件名組合成完整的路徑名。
四、總結
我們可以使用Python的OS模塊讀取目錄及其子目錄中的所有文件列表。可以通過調用os模塊提供的listdir
函數獲取指定目錄中的文件列表,或者使用遞歸遍歷來獲取一個目錄樹下的全部文件列表。
通過使用OS模塊和Python,可以輕鬆管理文件和目錄,使得文件管理變得更加簡單和高效。
原創文章,作者:小藍,如若轉載,請註明出處:https://www.506064.com/zh-hk/n/233665.html