在目录中搜索文件列表的最快方法

Question

给定一个文件名列表，如果这些文件存在于给定目录或其子目录中，我想为这些文件创建完整路径。 现在我用这段代码

def construct_path(file_list, directory):
    file_path_list = []
    for name in file_list:
        for dir, subdir, filenames in os.walk(directory):
            if name in filenames:
                file_path_list.append(os.path.join(dir, name))

    return file_path_list

因此，这里正在为列表中的每个文件搜寻目录。 有更快/更好的方法吗？

Answer 1

您可以删除file_list的循环并从此列表中删除找到的文件，因此一旦找到所有文件，便可以返回结果：

def construct_path(file_list, directory):
    file_path_list = []
    for dir, subdir, files in os.walk(directory):
        for name in files:
            if name in file_list:
                file_path_list.append(os.path.join(dir, name))
                file_list.remove(name)
            if (len(file_list)==0): return file_path_list
    return file_path_list