python - 如何从目录中读取数百个 csv 文件并对每个文件中的特定列求和？

Question

我的目录中有大约 650 个 csv 文件。每个文件都有三列，带有标题(#ch# ##kev## ###count###)，以及 2050 行。此处部分显示了其中一个文件图像。

    ch       kev     count
     0        0        0
     1        0        0
     2        0        0
     .        .        .
     .        .        .  
     .        .        .
    100       0        30
    101       0        70
     .        .        .
     .        .        .
    200       .        1000
    201       0        1037
     .        .        .
     .        .        .
    2050      0        0        

我想对列(#ch#) 值的特定范围(100 -200) 的列(###count###) 进行求和。我只能为单个 csv 文件编写程序，如下所示:

    import csv

    cr = csv.reader(open("D:\\Pythontest\\test1.csv", 'r'))
    cr.next()
    ['ch', 'kev', 'count']

    total = 0

    for row in cr:
        if 100 <= int(row[0]) <= 200:
           total += int(row[2])

    print total

但我不知道如何一次为所有文件编写程序。我想从每个文件中提取总和值并将它们放在一个单独的文件中。我使用的是 Python 2.7。

Answer 1

由于您有一个文件的工作程序，您所需要的只是一种在每个文件上使用该程序的方法。我建议执行以下操作:

• 了解如何使用 glob 模块列出所有 csv 文件，http://pymotw.com/2/glob/
• 将代码移至函数 http://www.tutorialspoint.com/python/python_functions.htm ，将文件名作为参数并返回该文件的总数。
• 循环遍历 glob 为您提供的每个文件并运行该函数，将总数加起来。

祝你好运，如果您在其中一个步骤中遇到问题，请随时发布更具体的新问题。