Python合并列表将唯一值隐藏为逗号

Question

我正在尝试使它起作用。

这是我的数据：

data.csv

id,fname,lname,education,gradyear,attributes
1,john,smith,mit,2003,qa
1,john,smith,harvard,207,admin
1,john,smith,ft,212,master
2,john,doe,htw,2000,dev

尝试使用此代码。 在Internet上找到了此代码，还没有完全理解。

from itertools import groupby
import csv
import pprint


t = csv.reader(open('data.csv'))
t = list(t)


def join_rows(rows):
    def join_tuple(tup):
        for x in tup:
            if x: 
                return x
        else:
            return x
    return [join_tuple(x) for x in zip(*rows)]



for name, rows in groupby(sorted(t), lambda x:x[0]):
    print join_rows(rows)

但是，它不会将唯一值合并为逗号分隔。

输出为：

['1', 'john', 'smith', 'ft', '212', 'master']
['2', 'john', 'doe', 'htw', '2000', 'dev']
['id', 'fname', 'lname', 'education', 'gradyear', 'attributes']

我如何使它像：

['1', 'john', 'smith', 'mit,harvard,ft', '2003,207,212', 'qa,admin,master']
['2', 'john', 'doe', 'htw', '2000', 'dev']
['id', 'fname', 'lname', 'education', 'gradyear', 'attributes']

如果同一列有更多条目，则它也应该起作用。 不应限于3行。

Grrrrr ....有人有提示或想法吗？

提前致谢！

Answer 1

您可以将join_rows的定义join_rows为

import itertools

def join_rows(rows):
    return [(e[0] if i < 3 else ','.join(e)) for (i, e) in enumerate(zip(*rows))]

这是将属于同一id的所有条目压缩到元组中。 对于前三个元组，返回第一项； 对于后者，它们之间以逗号分隔。

['1', 'john', 'smith', 'ft,harvard,mit', '212,207,2003', 'master,admin,qa']
['2', 'john', 'doe', 'htw', '2000', 'dev']
['id', 'fname', 'lname', 'education', 'gradyear', 'attributes']