0

我正在尝试创建一个脚本,该脚本将编写一个包含来自两个单独输入文件的列的 CSV 文件。我在尝试csv.Dictreader()在两个输入文件上都使用时遇到问题,以便可以轻松引用要写入的文件中的列。使用print,似乎两个字典中的键都被正确理解,但只有一个字典中的信息(曾经放在脚本中的第一个)被正确写入,而第二个只会写入每行中的第一个值。

输入文件的示例:

文件 1.csv

A,B
1,2
3,4
5,6

文件2.txt

1.11
2.22
3.33

我要创建的输出文件,第一列是 C,第二列是 A:

输出.csv

1.11     1
2.22     3
3.33     5

这是我到目前为止的代码:

import csv

with open('file1.csv', 'rb') as FileOne:
    with open('file2.txt', 'rb') as FileTwo
        with open('output.csv', 'wb') as OutFile:
            LabelsOne = ['A', 'B']
            LabelsTwo = ['C']
            inf.next()
            DictOne = csv.DictReader(FileOne, LabelsOne, skipinitialspace=True)
            DictTwo = csv.DictReader(FileTwo, LabelsTwo, skipinitialspace=False)
            outf = csv.writer(OutFile, delimiter='\t')

            for DataOne in DictOne:
                for DataTwo in DictTwo:
                    DataOne.update(DataTwo)
                    outf.writerow([DataOne['C'], DataOne['A']])

但现在我得到的输出是

1.11     1
2.22     1
3.33     1

或者

1.11    1
1.11    3
1.11    5

取决于我是先放for DataOne in DictOne还是for DataTwo in DictTwo先放。我查看了有关使用 CSV 字典的其他讨论,但没有发现其他人遇到此问题。两个字典中的键都被肯定地识别,但只有一个被正确写入。是否可以使用两个字典编写输出文件?

我知道打开 CSV 文件并将列复制/粘贴到新文档通常会更容易,但我需要这个过程重复许多文件并作为批处理的一部分。

4

1 回答 1

1

您根本不想要嵌套循环,请尝试使用zip()

            for DataOne, DataTwo in zip(DictOne, DictTwo):
                DataOne.update(DataTwo)
                outf.writerow([DataOne['C'], DataTwo['A']])
于 2013-10-29T23:32:54.183 回答