在Python中,通过os模块可以方便地遍历文件夹中的文件和子文件夹,但是如果我们需要遍历同一文件夹下的同类型文件,应该怎么做呢?例如,我们需要遍历一个文件夹下的所有.txt文件。
1. 使用os模块
可以使用os模块中的listdir方法获取文件夹下所有的文件和子文件夹。然后通过遍历获取到的文件和文件夹列表,筛选出符合条件的文件。
import os
path = 'folder_path'
files = os.listdir(path)
for file in files:
if file.endswith('.txt'):
print(os.path.join(path, file))
2. 使用glob模块
glob模块可以根据指定的通配符搜索匹配的文件路径,这样可以更方便地获取指定类型的文件路径。
import glob
path = 'folder_path/*.txt'
files = glob.glob(path)
for file in files:
print(file)
3. 使用Pathlib模块
Pathlib模块可以方便地进行文件路径的操作,包括遍历、过滤和获取文件类型等。
from pathlib import Path
path = Path('folder_path')
files = path.glob('*.txt')
for file in files:
print(file)
以上三种方法均可以达到遍历同类型文件的目的,选择最适合自己的方法即可。