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)从第一个文件到最后一个文件,而不是从最后一个文件到第一个文件.