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库文档](