在日常的编程工作中，我们经常需要读取文件夹中的特定文件。这对于数据分析、文件处理等任务来说尤为重要。Python作为一种强大而灵活的编程语言，提供了各种方法来实现这个目标。

使用os模块遍历文件夹

首先，我们需要使用Python的os模块来遍历文件夹。这个模块提供了一些函数，可以帮助我们获取文件夹中的所有文件和子文件夹。

通过调用os.listdir()函数，我们可以获取指定文件夹中所有的文件和文件夹的名称：

import os
def get_files_in_folder(folder_path):
    files = []
    for file_name in os.listdir(folder_path):
        files.append(file_name)
    return files

上述代码将返回一个包含文件夹中所有文件和文件夹名称的列表。

过滤文件

有时候，我们只需要获取文件夹中特定类型的文件。例如，我们只关心文件夹中的图片文件。在这种情况下，我们可以使用文件的扩展名来进行过滤。

import os
def get_files_with_extension(folder_path, extension):
    files = []
    for file_name in os.listdir(folder_path):
        if file_name.endswith(extension):
            files.append(file_name)
    return files

上述代码将返回一个仅包含指定扩展名文件的列表。

根据文件名过滤

除了根据文件扩展名过滤文件，我们还可以根据文件名中的特定字符进行过滤。这对于文件名具有一定规律性的情况下尤为有用。例如，我们只想读取文件名以”data_”开头的文件。

import os
def get_files_with_prefix(folder_path, prefix):
    files = []
    for file_name in os.listdir(folder_path):
        if file_name.startswith(prefix):
            files.append(file_name)
    return files

上述代码将返回一个仅包含以指定前缀开头的文件的列表。

使用正则表达式过滤

另一种灵活的方法是使用正则表达式来过滤文件。这种方法对于需要更复杂匹配规则的情况非常有用。

import os
import re
def get_files_with_pattern(folder_path, pattern):
    files = []
    for file_name in os.listdir(folder_path):
        if re.search(pattern, file_name):
            files.append(file_name)
    return files

上述代码将返回一个仅包含与指定正则表达式模式匹配的文件的列表。

总结

通过使用Python的os模块和一些简单的技巧，我们可以轻松地读取文件夹中部分的文件。无论是根据文件扩展名、文件名前缀，还是使用正则表达式来过滤，Python提供了灵活而高效的方法。这些技巧可以帮助我们更好地处理文件夹中的大量数据，并加快我们的编程工作。