Python - 使用字符串列表搜索文件夹名称

时间:2017-06-15 04:15:21

标签: python

我有一个字符串列表(存储在.txt文件中,每行一个字符串),我想创建一个脚本,它接受第一行并搜索目录中的所有文件夹名称,然后取第二行和搜索所有文件夹名称等。我该怎么做呢?希望我清楚自己。 THKS!

2 个答案:

答案 0 :(得分:1)

此示例从文本文件中读取路径并将其打印出来。用搜索逻辑替换打印件。

import os

textfile = open('C:\\folder\\test.txt', 'r')

for line in textfile:
    rootdir = line.strip()
    for subdir, dirs, files in os.walk(rootdir):
        for file in files:
            print(os.path.join(subdir, file))

答案 1 :(得分:1)

假设通过搜索所有文件夹意味着将它们打印到标准输出,您可以这样做:

from os import listdir
from os.path import isdir, join


with open('directories.txt', 'r') as f:
    i = 1
    for line in f.readlines():
        directories = []
        tmp = line.strip('\n')
        for d in listdir(tmp):
            if isdir(join(tmp, d)):
                directories.append(d)
        print('directory {}: {}'.format(i, directories))
        i += 1

它将输出如下内容:

directory 1: ['subfolder_1', 'subfolder_0']
directory 2: ['subfolder_0']
directory 3: []

请注意,我建议使用with来打开文件,因为即使发生异常,它也会自动正确关闭它们。