如何遍历文件夹以查找 Python 中的特定文件？

Question

我正在尝试遍历文件夹以对特定文件执行分析。

假设我有 3 个文件夹，分别命名为 S1、S2 和 S3。每个文件夹都包含一个同名文件，名为hi.csv。一旦我让脚本成功遍历每个文件夹并找到该文件，我将让它分析 CSV 并在每个文件夹中创建一个新的 CSV 文件。

代替分析代码，我只是尝试在第一个实例中打印文件。理想情况下，我会在每个子文件夹S1、S2 和S3 下看到一个名为hi.txt 的文件名。

这是我迄今为止尝试过的，但它不起作用：

fn = 'hi.txt'
indir = '/Users/sheena/Desktop/Test'
for root, dirs, filenames in os.walk(indir):
    for d in dirs:
        if os.path.isfile(fn):
            print(f)

Answer 1

如果您尝试检查某个文件名，则应在filenames 中检查，如果成功，则当前文件夹为root。 dirs 不进图。

fn = 'hi.txt'
indir = '/Users/sheena/Desktop/Test'
for root, _, filenames in os.walk(indir):
    if fn in filenames:
        print("Got: {}/{}".format(root, fn))

看看它是否按你想要的方式工作。如果没有请评论。

Answer 2

查看问题的简单方法是添加一行

print(root, dirs, filenames)

就在第一个“for”循环之后。您将得到的输出类似于

/tmp ['s3', 's2', 's1', ...] ['test.py', 'GRADUATE BALLOT APPLICATION FORM 2016-17.doc', 'ankid6955d9721560531274cb8f50ff595a9bd39d66f', '.X0-lock']
/tmp/s3 [] ['hi.txt']
/tmp/s2 [] ['hi.txt']
/tmp/s1 [] ['hi.txt']
/tmp/hsperfdata_joshua [] ['391']
/tmp/cujc ['foo'] []
/tmp/yaourt-tmp-joshua [] []
/tmp/.Test-unix [] []
/tmp/.font-unix [] []
/tmp/.XIM-unix [] []

所以

1. "os.walk" 可能没有按照您的预期执行，它正在遍历所有子目录和子目录的子目录（但这对您的代码来说不是问题，可能只是更智能/更快方法）
2. 所有文件都在“文件名”列表中 - 所以您遍历了错误的列表！当“root”是“blah/S1”、“blah/S2”或“blah/S3”时，“hi.txt”在“文件名”列表中。

Answer 3

if os.path.isfile(os.path.join(d,fn)):

我猜......也许......这可能不是找到匹配文件的最佳方法

for root, dirs, filenames in os.walk(indir):
    for fname in filenames:
        if fname == fn:
           print os.path.join(root,fname)

可能会好一点