Python获取指定格式文件路径|极客教程

Python获取指定格式文件路径

在Python编程中，有时我们需要获取特定格式的文件路径，对于一些需要批量处理特定格式文件的情况，这一功能就显得尤为重要。本文将详细介绍如何使用Python获取指定格式文件的路径。

os模块

在Python中，可以使用os模块来实现文件路径的操作。os模块是Python的标准库之一，提供了许多系统级的操作函数，包括文件和目录的操作。

os.listdir()方法

os.listdir()方法可以获取指定路径下的所有文件和目录，返回一个包含所有文件和目录名称的列表。这个方法是我们获取指定格式文件路径的基础。

import os

# 指定文件夹路径
folder_path = 'C:/Users/username/Documents/'

# 获取文件夹下所有文件和目录
files = os.listdir(folder_path)

# 输出所有文件和目录
for file in files:
    print(file)

运行以上代码，将打印出指定路径下的所有文件和目录名称。接下来，我们将扩展这个功能，获取特定格式的文件路径。

文件筛选

借助os.listdir()方法返回的文件列表，可以通过对文件名进行筛选来获取特定格式的文件路径。

import os

# 指定文件夹路径
folder_path = 'C:/Users/username/Documents/'

# 获取文件夹下所有文件和目录
files = os.listdir(folder_path)

# 指定文件格式
file_format = '.txt'

# 筛选特定格式的文件路径
for file in files:
    if file.endswith(file_format):
        file_path = os.path.join(folder_path, file)
        print(file_path)

在以上代码中，指定了文件格式为’.txt’，筛选出所有以’.txt’结尾的文件路径，并打印出来。可以根据实际需求修改文件格式。这样，我们就实现了获取特定格式文件路径的功能。

glob模块

除了os模块，还可以使用glob模块来获取指定格式文件的路径。glob模块提供了一个函数glob()，可以用来查找匹配指定模式的文件路径。

import glob

# 指定文件夹路径
folder_path = 'C:/Users/username/Documents/'

# 指定文件格式
file_format = '*.txt'

# 使用glob模块获取文件路径
files = glob.glob(folder_path + file_format)

# 打印文件路径
for file in files:
    print(file)

上述代码中，使用glob.glob()函数查找所有以’.txt’结尾的文件路径，并打印出来。同样，可以根据实际需求修改文件格式。