前言

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

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個(gè)內(nèi)容：絕對(duì)路徑、子文件夾、文件名。 此方法可以遍歷文件夾下的所有文件、子文件及內(nèi)的所有文件：

例如：

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

此方法在輸入的path為D:/workspace時(shí)，第一次遍歷會(huì)獲取/workspace下的所有文件和文件夾；第二次遍歷會(huì)獲取/workspace/folder下的所有文件和文件夾。因此需要對(duì)結(jié)果進(jìn)行循環(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ù)為路徑以及文件過(guò)濾條件，若不設(shè)置過(guò)濾需填寫(xiě)為*，此函數(shù)會(huì)返回包括路徑的文件夾和文件名

示例：

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

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

附：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))

總結(jié)