Python筛选需要的文件

在Python中，可以使用多种方法来筛选需要的文件。以下是几种常见的方法：

1. 使用os模块和文件名匹配：
可以使用os模块的相关函数（如os.listdir()）获取指定目录下的所有文件，然后使用字符串匹配方法（如str.endswith()）筛选需要的文件。

示例代码：

   import os

   def filter_files_by_extension(directory, extension):
       files = os.listdir(directory)
       filtered_files = [file for file in files if file.endswith(extension)]
       return filtered_files

   # 调用示例
   directory = '/path/to/directory'
   extension = '.txt'
   filtered_files = filter_files_by_extension(directory, extension)
   print(filtered_files)

2. 使用glob模块：
glob模块提供了一个更方便的方法来筛选文件，它支持通配符匹配。

示例代码：

   import glob

   def filter_files_by_pattern(directory, pattern):
       files = glob.glob(directory + pattern)
       return files

   # 调用示例
   directory = '/path/to/directory'
   pattern = '*.txt'
   filtered_files = filter_files_by_pattern(directory, pattern)
   print(filtered_files)

3. 使用fnmatch模块：
fnmatch模块提供了类似于shell中的通配符匹配功能，可以用来筛选文件。

示例代码：

   import os
   import fnmatch

   def filter_files_by_pattern(directory, pattern):
       files = os.listdir(directory)
       filtered_files = [file for file in files if fnmatch.fnmatch(file, pattern)]
       return filtered_files

   # 调用示例
   directory = '/path/to/directory'
   pattern = '*.txt'
   filtered_files = filter_files_by_pattern(directory, pattern)
   print(filtered_files)