python - 需要处理目录中的所有文件，但我只得到一个

Question

我有一个可以成功处理单个文件的 Python 脚本。我正在尝试编写一个 for 循环，让它获取用户输入的目录中的所有文件。但是，它只处理目录中的一个文件。

下面的代码后面是对数据 进行分析的其余脚本。我是否需要以某种方式关闭 for 循环？

import os

print "Enter the location of the files: "; directory = raw_input()
path = r"%s" % directory

for file in os.listdir(path):
    current_file = os.path.join(path, file)

    data = open(current_file, "rb")

    # Here's an abridged version of the data analysis

    for i in range(0, 10):
        fluff = data.readline()

    Input_Parameters = fluff.split("\t")

    output.write("%f\t%f\t%f\t%f\t%.3f\t%.1f\t%.2f\t%.2f\t%s\n" % (Voc, Isc, Vmp, Imp, Pmax, 100 * Eff, IscErr, 100 * (1 - (P2 / Pmax)), file))
    data.close()

Answer 1

一般来说，如果某些东西不起作用，您可以尝试让一些更简单的东西起作用。我删除了数据分析部分并保留了下面的代码。这对我有用。我注意到如果我的路径中有一个目录，打开将失败。我不确定你也是这种情况。

import os
import sys

path = '.'

for file in os.listdir(path):
    current = os.path.join(path, file)
    if os.path.isfile(current):
        data = open(current, "rb")
        print len(data.read())