python - 如何合并两个csv文件？

Question

我有两个这样的 csv 文件

"id","h1","h2","h3", ...
"1","blah","blahla"
"4","bleh","bleah"

我想合并这两个文件，这样如果两个文件中的 ID 相同，则该行的值应该来自第二个文件。如果它们具有不同的 ID，则合并后的文件应包含这两行。

---

有些值有逗号

"54","34,2,3","blah"

Answer 1

res = {}

a=open('a.csv')
for line in a:
    (id, rest) = line.split(',', 1)
    res[id] = rest
a.close()

b=open('b.csv')
for line in b:
    (id, rest) = line.split(',', 1)
    res[id] = rest
b.close()

c=open('c.csv', 'w')
for id, rest in res.items():
    f.write(id+","+rest)
f.close()

基本上，您使用每行的第一列作为字典 res 中的键。因为 b.csv 是第二个文件，所以第一个文件 (a.csv) 中已经存在的 key 将被覆盖。最后，您在输出文件 c.csv 中再次将 key 和 rest 合并在一起。

标题行也将从第二个文件中获取，但我猜这些应该没有什么不同。

编辑:一个略有不同的解决方案，合并任意数量的文件并按顺序输出行:

res = {}
files_to_merge = ['a.csv', 'b.csv']
for filename in files_to_merge:
    f=open(filename)
    for line in f:
        (id, rest) = line.split(',', 1)
        if rest[-1] != '\n': #last line may be missing a newline
            rest = rest + '\n'
        res[id] = rest
    f.close()

f=open('c.csv', 'w')
f.write("\"id\","+res["\"id\""])
del res["\"id\""]
for id, rest in sorted(res.iteritems()):
    f.write(id+","+rest)
f.close()