如何遍历列表结合glob.glob匹配指定规则的.pkl文件
解决方案
你的问题核心是没有正确遍历listing生成匹配模式,也没有替换占位符为实际的字符串值。原代码里的*{x}_.pkl只是字面量字符串,不会自动替换为listing里的元素,且只执行了一次glob.glob,自然无法匹配到目标文件。
下面是两种可行的实现方式:
方法一:循环遍历+逐个匹配(直观易读)
import glob import os listing = ['DBMP','CIFP'] input_location = "/你的实际输入路径" # 替换为真实路径 pklFilenamesList = [] # 遍历listing中的每个字符串 for item in listing: # 用f-string生成对应匹配规则:*DBMP_.pkl、*CIFP_.pkl match_pattern = os.path.join(input_location, f"*{item}_.pkl") # 将当前匹配到的文件列表合并到总列表中 pklFilenamesList.extend(glob.glob(match_pattern)) # 遍历结果示例 for filecounter, filename in enumerate(pklFilenamesList): print(f"序号{filecounter}: {filename}")
方法二:列表推导式(简洁写法)
如果追求代码简洁,可以用嵌套列表推导式一次性生成结果:
import glob import os listing = ['DBMP','CIFP'] input_location = "/你的实际输入路径" pklFilenamesList = [ file for item in listing for file in glob.glob(os.path.join(input_location, f"*{item}_.pkl")) ] # 遍历结果 for filecounter, filename in enumerate(pklFilenamesList): print(f"序号{filecounter}: {filename}")
关键说明
- 用f-string(
f"*{item}_.pkl")实现字符串替换,把listing中的每个元素插入到匹配规则里; - 必须遍历
listing的每个元素,分别调用glob.glob,再将结果合并,才能覆盖所有目标匹配规则; - 如果存在重复文件名的情况,可以将结果转为集合去重后再转回列表:
pklFilenamesList = list(set(pklFilenamesList))。
内容的提问来源于stack exchange,提问作者Jonnyboi
相关产品推荐
相关产品推荐

