Python：如何在目录名称(不是单个文件名)中搜索特定"string"

Question

我想创建一个与特定字符串匹配的所有文件路径名称的列表，例如"04_DEM"，以便我可以对这些目录中的文件进行进一步处理？

例如

输入

 C:directoryNewZealand4DEMDEM_CD23_1232.tif
 C:directoryAustralia4DEMDEM_CD23_1233.tif
 C:directoryNewZealand5DSMDSM_CD23_1232.tif
 C:directoryAustralia5DSMDSM_CD23_1232.tif

想要的输出

 C:directoryNewZealand4DEM
 C:directoryAustralia4DEM

这可确保仅处理这些文件，因为目录中的其他一些文件也在其文件名中包含相同的字符串"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:directoryNewZealand4DEMDEM_CD23_1232.tif',
       r'C:directoryAustralia4DEMDEM_CD23_1233.tif',
       r'C:directoryNewZealand5DSMDSM_CD23_1232.tif',
       r'C:directoryAustralia5DSMDSM_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']

Answer 2

使用 in 检查所需的字符串是否在另一个字符串中。

这是一种快速方法：

new_list = []
for path in path_list:
    if '04DEM' in path:
        new_list.append(path)

演示：

s = 'C:/directory/NewZealand/04DEM/DEM_CD23_1232.tif'
if '04DEM' in s:
    print(True)
# True

请确保使用 / 或 \ 作为目录分隔符而不是 ，因为后者会转义字符。

Answer 3

首先，使用 re 通过正则表达式进行选择，然后使用 pathlib ：

import re
import pathlib
pattern = re.compile('04DEM')
# You use pattern.search() if s is IN the string
# You use pattern.match() if s COMPLETELY matches the string.
# Apply the correct function to your use case.
files = [s in list_of_files if pattern.search(s)]
all_pruned_paths = set()
for p in files:
    total = ""
    for d in pathlib.Path(p):
        total = os.path.join(total, d)
        if pattern.search(s):
            break
    all_pruned_paths.add(total)
result = list(all_pruned_paths)

这比使用 in 更可靠，因为将来可能需要形成更复杂的查询。