极客笔记Python读取文件夹中的文件名
在编程中，有时我们需要读取文件夹中的所有文件名，这在处理批量操作时非常有用。Python提供了一些库和方法来实现这个功能。在本文中，我们将详细讨论如何使用Python来读取文件夹中的文件名。
使用os库
Python的os模块提供了一种简单的方法来读取文件夹中的文件名。我们首先需要导入os模块，然后使用os.listdir()方法来获取文件夹中的所有文件名。下面是一个示例代码：
import os
folder_path = 'path_to_your_folder'
file_names = os.listdir(folder_path)
for file_name in file_names:
print(file_name)
上面的代码中，我们首先定义了一个文件夹路径folder_path，然后使用os.listdir()方法获取该文件夹中的所有文件名，并将文件名逐个打印出来。
使用glob库
除了os模块，Python的glob库也是一个非常方便的工具来读取文件夹中的文件名。使用glob.glob()方法可以同时筛选指定后缀的文件。下面是一个示例代码：
import glob
folder_path = 'path_to_your_folder'
file_names = glob.glob(folder_path + '/*.txt') # 只获取txt后缀的文件名
for file_name in file_names:
print(file_name)
在上面的代码中，我们首先通过glob.glob()方法获取指定后缀的文件名列表，并将其打印出来。
使用os.walk()方法
除了os和glob库，Python的os模块还提供了一个os.walk()方法，该方法可以遍历文件夹及其子文件夹中的所有文件。下面是一个示例代码：
import os
folder_path = 'path_to_your_folder'
for root, dirs, files in os.walk(folder_path):
for file_name in files:
print(os.path.join(root, file_name))
在上面的代码中，我们使用os.walk()方法遍历文件夹及其子文件夹中的所有文件，并将它们的完整路径打印出来。
运行结果
假设我们有一个文件夹test_folder，其中包含以下文件：
- file1.txt
- file2.txt
- sub_folder/file3.txt
我们可以使用上面的示例代码来读取文件夹中的文件名。假设文件夹路径是/home/user/test_folder，那么运行以上代码的输出应该如下：
file1.txt
file2.txt
sub_folder/file3.txt
通过以上示例代码，我们可以轻松地读取文件夹中的所有文件名，为之后的批量操作提供便利。