Python：如何在目录名（而不是单个文件名）中搜索特定的“字符串”

Question

我想创建一个列表，列出所有与特定字符串匹配的文件路径名，例如“04\u DEM”，以便对这些目录中的文件进行进一步处理？你知道吗

例如

输入

 C:\directory\NewZealand\04DEM\DEM_CD23_1232.tif
 C:\directory\Australia\04DEM\DEM_CD23_1233.tif
 C:\directory\NewZealand\05DSM\DSM_CD23_1232.tif
 C:\directory\Australia\05DSM\DSM_CD23_1232.tif

想要的产出

 C:\directory\NewZealand\04DEM\
 C:\directory\Australia\04DEM\

这确保只处理这些文件，因为目录中的其他一些文件的文件名中也包含相同的字符串“DEM”，我不想修改它。你知道吗

这是我的错误尝试，因为我是一个有Py代码的新手

 import os

 for dirnames in os.walk('D:\Canterbury_2017Copy'):
     print dirnames
     if dirnames=='04_DEM' > listofdirectoriestoprocess.txt

 print "DONE CHECK TEXT FILE"

Answer 1

您可以使用os.path来实现：

import os

lst = [r'C:\directory\NewZealand\04DEM\DEM_CD23_1232.tif',
       r'C:\directory\Australia\04DEM\DEM_CD23_1233.tif',
       r'C:\directory\NewZealand\05DSM\DSM_CD23_1232.tif',
       r'C:\directory\Australia\05DSM\DSM_CD23_1232.tif']

def filter_paths(lst, x):
    return [os.path.split(i)[0] for i in lst if os.path.normpath(i).split(os.sep)[3] == x]

res = list(filter_paths(lst, '04DEM'))

# ['C:\\directory\\NewZealand\\04DEM',
#  'C:\\directory\\Australia\\04DEM']