利用Python獲取文件夾下所有文件實例代碼

更新時間：2023年01月04日 11:04:32 作者：琲世

在處理數(shù)據(jù)的過程中經(jīng)常需要遍歷文件夾,如果遠程服務器的文件是分布式存儲,遍歷需要更快的速度,下面這篇文章主要給大家介紹了關于利用Python獲取文件夾下所有文件的相關資料,需要的朋友可以參考下

前言

使用Python獲取文件夾下的所有文件時，存在多種方式。

1. os.listdir

os.listdir：參數(shù)為文件夾路徑，可以返回文件夾下的所有子文件夾、文件名稱。

示例：

import os

path = 'D:\Workspace'
for file_name in os.listdir(path):
    print(file_name)

注意，不能返回子文件夾下的文件，并且僅返回名稱。

2. os.walk

os.walk：參數(shù)為文件夾路徑，返回3個內(nèi)容：絕對路徑、子文件夾、文件名。此方法可以遍歷文件夾下的所有文件、子文件及內(nèi)的所有文件：

例如：

D:/workspace/folder/text1.txt
D:/workspace/folder/text2.txt

此方法在輸入的path為D:/workspace時，第一次遍歷會獲取/workspace下的所有文件和文件夾；第二次遍歷會獲取/workspace/folder下的所有文件和文件夾。因此需要對結果進行循環(huán)，以獲取所有文件。

示例：

import os

path = 'D:\Workspace'
for root, dirs, files in os.walk(path):
    print(root)
    print(dirs)
    print(files)

返回：

D:\Workspace
['folder']
[]
D:\Workspace\folder
[]
['text1.txt', 'text2.txt']

3. glob.glob

glob：參數(shù)為路徑以及文件過濾條件，若不設置過濾需填寫為*，此函數(shù)會返回包括路徑的文件夾和文件名

示例：

import glob

path = 'D:\Workspace\folder\*'
for file_abs in glob.glob(path):
    print(file_abs)

返回：

D:\Workspace\folder\text1.txt
D:\Workspace\folder\text2.txt

注意，不能返回子文件夾下的文件，并且返回文件名與絕對路徑。

附：Python獲取指定目錄下的文件夾名列表和文件名列表

import os
 
# enter of program
if __name__ == '__main__':
    # input
    input_path = 'D:/data/pycodes/books/'
 
    # initialize
    dir_list = []
    file_list = []
 
    # process
    for root, dirs, files in os.walk(input_path):
        dir_list = dirs
        file_list = files
        break
 
    # output
    print(dir_list, len(dir_list))
    print(file_list, len(file_list))