我正在尝试创建一个脚本,该脚本将编写一个包含来自两个单独输入文件的列的 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 文件并将列复制/粘贴到新文档通常会更容易,但我需要这个过程重复许多文件并作为批处理的一部分。