Как объединить несколько строк в CSV в одну строку

Question

Мне был предоставлен большой файл CSV, который мне нужно разбить для использования в машинах. Мне удалось найти способ разделения файла на две строки, которые мне нужны, но у меня есть проблема.

У меня в основном такая файловая структура.

"David", "Red" 
"David", "Ford" 
"David", "Blue" 
"David", "Aspergers" 
"Steve", "Red" 
"Steve", "Vauxhall" 

И я требую, чтобы данные выглядеть следующим образом ...

"David, "Red", "Ford", "Blue", "Aspergers" 
"Steve", "Red", "Vaxhaull" 

я в настоящее время это обирать файлы CSV

import csv 

cr = csv.reader(open("traits.csv","rb"), delimiter=',', lineterminator='\n') 
cr.next() #skipping header line, no point in removing it as I need to standardise data manipuation. 


# Print out the id of species and trait values 
print 'Stripping input' 
vals = [(row[1], row[4]) for row in cr] 
print str(vals) + '\n' 

with open("output.csv", "wb") as f: 
    writer = csv.writer(f) 
    writer.writerows(vals) 
    print 'Sucessfully written to file output.csv' 


#for row in cr: 
#print row 

Answer 1

Используйте словарь для хранения имен в качестве ключа и других атрибутов в списке в качестве значения:

my_dict={} 
with open("traits.csv","rb") as f: 
    cr = csv.reader(f, delimiter=',', lineterminator='\n') 
    for row in cr: 
     my_dict.setdefault(row[0].strip('" '),[]).append(row[1].strip('" ')) 

результат:

print my_dict 
{'Steve': ['Red', 'Vauxhall'], 'David': ['Red', 'Ford', 'Blue', 'Aspergers']} 

И для записи в новом файле:

with open("output.csv", "wb") as f: 
    writer = csv.writer(f,delimiter=',') 
    for i,j in my_dict.iteritems(): 
     writer.writerow([i]+j) 

setdefault(key[, default])

If key is in the dictionary, return its value. If not, insert key with a value of default and return default. default defaults to None.

Answer 2

Использование defaultdict, это именно то, что вам нужно, вот пример:

>>> from collections import defaultdict 
>>> md = defaultdict(list) 
>>> md[1].append('a') 
>>> md[1].append('b') 
>>> md[2].append('c') 
>>> md[1] 
['a', 'b'] 
>>> md[2] 
['c'] 

(Вместо списка вы можете использовать set, в этом случае вы, d call .add вместо .append.)

Вы можете использовать iteritems для легкого доступа к данным.