Python获取文件夹下所有文件的方法详解

Python能够轻松地遍历文件夹下的所有文件，下面从不同的角度，详细地讲述如何在Python中获取文件夹下所有文件。

一、使用os库遍历文件夹

Python中的os库提供了许多底层的操作函数，包括文件访问、目录操作、进程管理等等。遍历文件夹中的所有文件可以使用os库中的os.listdir()函数。

import os

def get_all_files(rootdir):
    fileList = []
    for root, dirs, files in os.walk(rootdir):
        for file in files:
            filepath = os.path.join(root, file)
            fileList.append(filepath)
    return fileList

rootdir = "/path/to/folder"
fileList = get_all_files(rootdir)
for file in fileList:
    print(file)

首先定义了一个函数get_all_files()，接受文件夹路径，返回文件路径列表。os.walk()函数用来返回三元组(root,dirs,files)，其中root表示正在遍历的当前文件夹路径，dirs表示当前路径下所有文件夹列表，files表示当前路径下所有文件列表。

通过os.path.join()函数，将文件夹路径和文件名连接起来，获取文件的完整路径。将所有的文件路径添加到一个列表中，最后返回。

二、使用glob库遍历文件夹

glob库是Python自带的一个文件路径匹配工具，可以通过匹配规则获取文件路径。使用它可以非常方便地遍历文件夹中的所有文件。

import glob

def get_all_files(rootdir):
    fileList = []
    pattern = rootdir + '/**/*'
    files = glob.glob(pattern, recursive=True)
    for file in files:
        if not os.path.isdir(file):
            fileList.append(file)
    return fileList

rootdir = "/path/to/folder"
fileList = get_all_files(rootdir)
for file in fileList:
    print(file)

首先定义了一个函数get_all_files()，接受文件夹路径，返回文件路径列表。使用glob.glob()函数获取符合规则的文件列表，其中pattern表示文件路径匹配规则，recursive=True表示递归获取所有文件。

由于glob.glob()函数还会获取文件夹路径，因此需要额外判断文件是否为文件夹，将所有的文件路径添加到一个列表中，最后返回。

三、使用pathlib库遍历文件夹

pathlib库是Python3.4开始新增的库，用于简化与文件系统路径相关的操作。使用该库可以非常方便地遍历文件夹中的所有文件。

from pathlib import Path

def get_all_files(rootdir):
    fileList = []
    for path in Path(rootdir).rglob('*'):
        if path.is_file():
            fileList.append(str(path))
    return fileList

rootdir = "/path/to/folder"
fileList = get_all_files(rootdir)
for file in fileList:
    print(file)

首先定义了一个函数get_all_files()，接受文件夹路径，返回文件路径列表。使用Path().rglob()方法获取符合规则的文件列表，其中’*’表示匹配所有文件，rglob()表示递归获取所有文件。

由于Path().rglob()方法返回的是pathlib.Path对象，需要使用str()方法将其转换为字符串，最后判断文件是否为文件，将所有的文件路径添加到一个列表中，最后返回。

四、使用os.scandir()替代os.listdir()

在Python 3.5中，os库新增了一个os.scandir()函数，该函数与os.listdir()函数类似，用于遍历文件夹中的所有文件。相比于os.listdir()，os.scandir()可以提供更好的性能，尤其是在遍历大量文件时。

def get_all_files(rootdir):
    fileList = []
    for entry in os.scandir(rootdir):
        if entry.is_file():
            fileList.append(entry.path)
        elif entry.is_dir():
            sublist = get_all_files(entry.path)
            fileList.extend(sublist)
    return fileList

rootdir = "/path/to/folder"
fileList = get_all_files(rootdir)
for file in fileList:
    print(file)

首先定义了一个函数get_all_files()，接受文件夹路径，返回文件路径列表。使用os.scandir()函数获取文件夹中所有文件和文件夹的iterable，使用entry.path获取文件绝对路径。

由于需要递归获取所有文件，需要在遇到文件夹时，递归调用get_all_files()函数获取子文件夹中的所有文件，最后将所有的文件路径添加到一个列表中，最后返回。

五、结语

本文介绍了四种获取文件夹下所有文件的方法，使用os库遍历文件夹、使用glob库遍历文件夹、使用pathlib库遍历文件夹以及使用os.scandir()替代os.listdir()，作者个人认为最为推荐使用的是pathlib库的方法，因为它提供了非常方便的面向对象的文件系统操作方式，同时也更加Pythonic。

原创文章，作者：BJPK，如若转载，请注明出处：https://www.506064.com/n/136417.html