2016-12-06 3 views

答えて

1

私はあなたの入力ファイルがid_input.csvと呼ばれていると仮定します。

id_input.csv:ご希望の出力を印刷します

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

スクリプトは、次のとおりです。

だから、私の溶液中で、私は、この例のようにitertoolsモジュールからgroupbyを使用します

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