Python-使用字符串列表搜索文件夾名稱

Question

我有一個字符串列表（存儲在.txt文件中，每行一個字符串），我想制作一個腳本，該腳本使用第一行並搜索目錄中的所有文件夾名稱，然后使用第二行並搜索所有文件夾名稱等。 我該怎么做呢？ 希望我能說清楚。 ks！

Answer 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))

Answer 2

假設通過搜索所有文件夾意味着將它們打印到標准輸出中，則可以執行以下操作：

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來打開文件，因為即使發生異常，它也會自動正確關閉文件。