python 各种遍历文件夹的方法

Python 提供了一些库,可以帮助我们实现文件夹遍历。在本文中,我将介绍三种常用的遍历文件夹的方法,并且给出每种方法的使用示例和优缺点分析。

方法一:os.walk()

 os.walk() 是 Python 中最常用的遍历文件夹的方法。这个方法会自动地递归遍历文件夹中的所有子文件夹。在遍历过程中,返回的是一个元组(root, dirs, files),其中 root 表示当前遍历到的文件夹的路径,dirs 表示当前文件夹下的子文件夹,files 表示当前文件夹下的所有文件。

import os

# 遍历文件夹
def traversal_folder(folder_path):
    for root, dirs, files in os.walk(folder_path):
        # 遍历当前文件夹下的所有文件
        for file_name in files:
            file_path = os.path.join(root, file_name)
            print(file_path)

# 调用函数
folder_path = r"D:\data"
traversal_folder(folder_path)

优点:os.walk() 能够自动遍历文件夹下的子文件夹,无需手动递归。

缺点:对于大型文件夹,os.walk() 会占用大量的内存,因为它需要把所有的文件都存储在内存中,会导致程序崩溃。

方法二:os.listdir()

os.listdir() 是 Python 中另一种常用的遍历文件夹的方法。这个方法会返回指定路径下的所有文件和文件夹的列表,但是不会递归遍历。

import os

# 遍历文件夹
def traversal_folder(folder_path):
    for file_name in os.listdir(folder_path):
        file_path = os.path.join(folder_path, file_name)
        print(file_path)

# 调用函数
folder_path = r"D:\data"
traversal_folder(folder_path)

优点:os.listdir() 执行速度比 os.walk() 更快。

缺点:os.listdir() 不会递归遍历,如果需要递归遍历的话,还需要在函数中加入递归操作。

方法三:glob.glob()

glob.glob() 是 Python 中一种使用比较少见的遍历文件夹的方法。这个方法会根据指定的文件名模式匹配到的所有文件和文件夹,并以一个列表的形式返回。类似于 os.listdir(),不会递归遍历。

import glob

# 遍历文件夹
def traversal_folder(folder_path):
    for file_path in glob.glob(os.path.join(folder_path, '*')):
        print(file_path)

# 调用函数
folder_path = r"D:\data"
traversal_folder(folder_path)

优点:glob.glob() 使用简单,代码量少。

缺点:glob.glob() 不会递归遍历,如果需要递归遍历的话,还需要在函数中加入递归操作。

结论

       以上就是三种常用的 Python 遍历文件夹的方法。在实际使用中,我们可以根据具体情况选择不同的方法来实现文件夹的遍历。如果需要递归遍历,建议使用 os.walk() 方法,如果只需要遍历当前文件夹,建议使用 os.listdir() 或 glob.glob() 方法。

 

你可能感兴趣的:(python,开发语言)