我正在尝试将几行 csv 数据合并为一长行,前提是两个单元格包含相同的数据。例如,采用以下 csv:
one, two, three
1, 2, 3
4, 5, 6
7, 8, 9
1, 1, 1
4, 4, 4
如果两行在 row[0] 处共享相同的值,我希望将第二行附加到第一行。所以我的最终产品应该是这样的:
one, two, three
1, 2, 3, 1, 1, 1
4, 5, 6, 4, 4, 4
7, 8, 9
到目前为止,这是我的尝试:
import csv
uniqueNum = []
uniqueMaster = []
count = -1
with open("Test.csv", "rb") as source:
reader = csv.reader(source)
header = next(reader)
for row in reader:
if row[0] not in uniqueNum:
uniqueMaster.append(row)
uniqueNum.append(row[0])
count = count + 1
for row in reader:
if row[0] in uniqueNum:
uniqueMaster[count].append(row)
with open("holding.csv","wb") as result:
writer = csv.writer(result)
writer.writerow(header)
for row in uniqueMaster:
writer.writerow(row)
对我来说一切看起来都不错,但我的脚本仅输出以下内容:
one, two, three
1, 2, 3, ['1', '1', '1']
这显然是错误的,原因有两个。首先,它不会迭代整个 csv,其次,附加值被压缩到一个单元格中,而不是单个单元格中。如果有人对使其正常工作有任何建议,我将非常感激!