Python遍历文件夹读取文件名

在日常开发中,经常会遇到需要遍历文件夹并读取文件名的情况。Python作为一门功能强大的编程语言,提供了许多库和方法来实现这个功能。在本文中,我们将介绍如何使用Python来遍历文件夹并读取文件名,并且通过代码示例来演示具体的实现方法。

使用os库遍历文件夹

Python中的os库提供了许多与操作系统交互的方法,包括遍历文件夹的功能。我们可以使用os.walk()方法来遍历文件夹,并获取文件名。

import os

def list_files(directory):
    for root, dirs, files in os.walk(directory):
        for file in files:
            print(file)

# 调用函数,传入要遍历的文件夹路径
list_files("/path/to/directory")

上面的代码中,list_files()函数接收一个文件夹路径作为参数,然后使用os.walk()方法遍历该文件夹,并通过循环打印出文件名。

使用glob库遍历文件夹

除了os库之外,Python中的glob库也提供了一种简单的方法来遍历文件夹。我们可以使用glob.glob()方法来获取文件夹下的所有文件,并获取文件名。

import glob

def list_files(directory):
    files = glob.glob(directory + "/*")
    for file in files:
        print(os.path.basename(file))

# 调用函数,传入要遍历的文件夹路径
list_files("/path/to/directory")

上面的代码中,list_files()函数接收一个文件夹路径作为参数,然后使用glob.glob()方法获取该文件夹下的所有文件,并通过循环打印出文件名。

结合os.path.splitext()获取文件扩展名

有时候我们不仅需要获取文件名,还需要获取文件的扩展名。可以使用os.path.splitext()方法来实现这一功能。

import os

def list_files(directory):
    for root, dirs, files in os.walk(directory):
        for file in files:
            filename, extension = os.path.splitext(file)
            print(f"File: {filename}, Extension: {extension}")

# 调用函数,传入要遍历的文件夹路径
list_files("/path/to/directory")

总结

通过本文的介绍,我们学习了如何使用Python来遍历文件夹并读取文件名。我们使用了os库和glob库提供的方法,分别实现了遍历文件夹的功能。另外,我们还结合了os.path.splitext()方法来获取文件的扩展名。这些方法可以帮助我们在开发过程中快速、高效地处理文件操作。

希望本文对大家有所帮助,也希望大家能够在实际开发中灵活运用这些方法。让我们一起提升Python编程技能,更加高效地实现自己的想法和项目!

pie
    title 文件类型分布
    "txt" : 30
    "doc" : 20
    "pdf" : 10
    "jpg" : 40
stateDiagram
    [*] --> ListFiles
    ListFiles --> GetFileName
    GetFileName --> GetExtension
    GetExtension --> [*]

参考资料

  • [Python官方文档](
  • [Python glob库文档](