用 python 合并 csv 文件。 1 个 ID，多个条目

Question

我需要能够通过重复 id 运行 csv，并在每个 id 的一行上输出具有非重复值的 id。一直在尝试使用 pandas 来读取 csv。

我有什么：

Id, Bin  
1, a1  
1, a2  
1, b2  
2, a4  
2, a5  
2, b4  
2, b5

我想要什么：

Id, Bin  
1, a1, a2, b2  
2, a4, a5, b4, b5

Answer 1

我想您的输入文件名为id_input.csv。

因此，在我的解决方案中，我将使用来自itertools 模块的groupby，如下例所示：

id_input.csv：

Id, Bin  
1, a1  
1, a2  
1, b2  
2, a4  
2, a5  
2, b4  
2, b5

将打印所需输出的脚本是：

from itertools import groupby

data = list(k.replace("\n", "").strip().split(",") for k in open("id_input.csv", 'r'))


for key, val in groupby(data, lambda x : x[0]):
    if key != "Id":
        print("%s, %s" %(key, ",".join(k[1] for k in list(val))))

输出：

1,  a1, a2, b2
2,  a4, a5, b4, b5