Python：通过几个csv文件循环

Question

我想知道是否有人知道我可以如何在Python中更改脚本，以便它通过包含csv文件的文件夹，并将它们分成三组。当我在命令行中键入文件名时，脚本正在工作，但是我有很多文件，所以这会花费很长时间。它看起来像这样：

resultsdir = "blah" 

#filename1=sys.argv[1] 
#filename2=sys.argv[2] 
#filename3=sys.argv[3] 

file1 = open(resultsdir+"/"+filename1+".csv") 
file2 = open(resultsdir+"/"+filename2+".csv") 
file3 = open(resultsdir+"/"+filename3+".csv") 

我是一个完整的初学者，我希望我已经能够解释我想要的。欢呼任何帮助！

Answer 1

您可以使用glob模块（http://docs.python.org/3.3/library/glob.html）获取目录中的所有.csv文件，然后打开它们。

例子：

import glob 
resultsdir = "blah" 

files = sorted(glob.glob(resultsdir+'/*.csv')) 
while len(files) >= 3: 
    file1 = open(files.pop(0)) 
    file2 = open(files.pop(0)) 
    file3 = open(files.pop(0)) 
    # Do something 
# if the number of files can't be divided by 3 do something 
# with the 1 or 2 files which are left 

编辑：改变files.pop()到files.pop(0)拿到文件从第一个到最后，而不是从上次的第一个文件。

Answer 2

如果你想要的是一群由名单的三个要素，这里是做它的代码示例：

import itertools 


def groupby_three(iterable): 
    # x[0] is the index of the scanned element in the input list 
    for _, values in itertools.groupby(enumerate(iterable), 
             lambda x: x[0]/3): 
     yield([y[1] for y in values]) 

# Group by 3 the integers from 10 to 19 
for x in groupby_three(xrange(10, 20)): 
    print x 

输出：

[10, 11, 12] 
[13, 14, 15] 
[16, 17, 18] 
[19]