循环遍历多个.csv文件python / pandas

时间:2015-08-03 07:40:00

标签: python csv pandas

我有两个包含+50 .csv文件的文件夹,我想用pandas处理我的python代码中的那些文件。在我的代码的开头,我加载了两个不同的.csv文件:

Location1 = path\tasks_01.csv'          
Location2 = path\resource_01.csv'          

dftask = pd.read_csv(Location1)
dfresource = pd.read_csv(Location2)

在中间我做了各种不同的操作来构建数据等。最后我将两个.csv文件保存到一个新的.csv文件中:

dftask.to_csv(path\tasks_new.csv')
dfresource.to_csv(path\resource_new.csv')

由于我有两个文件夹,一个包含task.csv文件,另一个包含resource.csv文件,如何以这种方式编辑我的代码,我可以遍历所有这些文件?并以原来的名义保存它们?

希望你能帮助我!

1 个答案:

答案 0 :(得分:0)

在每个文件夹中创建一个文件列表,然后压缩它们。

import os

files_in_folder_1 = [os.path.join(path1, f) for f in os.listdir(path1) if os.path.isfile(os.path.join(path1, f))]

files_in_folder_2 = [os.path.join(path2, f) for f in os.listdir(path2) if os.path.isfile(os.path.join(path2, f))]

for file1, file2 in zip(files_in_folder_1, files_in_folder_2):
    with open(file1) as f1, open(file2) as f2:
        ...