小编Imr*_*an 的帖子

我正在试图弄清楚如何让python遍历一个充满csv文件的目录,处理每个文件并吐出带有修剪的值列表的文本文件.

在这个例子中,我正在迭代一个包含许多不同类型列的CSV,但我真正想要的只是名字,姓氏和关键字.我有一个充满了这些具有不同列的csv的文件夹(除了它们在csv中共享名字,姓氏和关键字).打开该文件夹的最佳方法是什么,浏览每个csv文件,然后将其全部吐出为自己的csv文件,仅用于文本列表,如下例所示.

import csv
reader = csv.reader(open("keywords.csv"))
rownum = 0
headnum = 0
F = open('compiled.txt','w')
for row in reader:
    if rownum == 0:
        header = row;
        for col in row:
            if header[headnum]=='Keyword':
                keywordnum=headnum;
            elif header[headnum]=='First Name':
                firstnamenum=headnum;
            elif header[headnum]=='Last Name':
                lastnamenum=headnum;
            headnum +=1
    else:
        currentrow=row
        print(currentrow[keywordnum] + '\n' + currentrow[firstnamenum] + '\n' + currentrow[lastnamenum]) 
        F.write(currentrow[keywordnum] + '\n')

    rownum +=1