在Python中,可以使用以下方法来挑选特定目录下的文件:
os
模块来遍历目录下的所有文件和文件夹:import os
def get_files_in_directory(directory, extension):
files = []
for root, dirs, filenames in os.walk(directory):
for filename in filenames:
if filename.endswith(extension):
files.append(os.path.join(root, filename))
return files
上述代码中,directory
参数表示要遍历的目录路径,extension
参数表示要挑选的文件扩展名。该函数会返回一个包含所有符合条件的文件路径的列表。
glob
模块来匹配特定的文件路径:import glob
def get_files_with_extension(directory, extension):
pattern = os.path.join(directory, f"*.{extension}")
files = glob.glob(pattern)
return files
上述代码中,directory
参数表示要搜索的目录路径,extension
参数表示要挑选的文件扩展名。该函数会返回一个包含所有符合条件的文件路径的列表。
这两种方法都可以根据特定的文件扩展名来挑选文件。你可以根据实际需求选择其中一种方法来使用。
腾讯云相关产品和产品介绍链接地址:
领取专属 10元无门槛券
手把手带您无忧上云