Добавление определенных полей из csv

Question

Я пытаюсь добавить определенные значения из файла CSV, если пользователь тот же. Я не могу объяснить это четко, поэтому я постараюсь показать вам.

===================== 
|E-mail | M-count | 
|a@a.com | 12  | 
|b@a.com | 8  | 
|a@a.com | 13  | 
|c@a.com | 2  | 
===================== 

Затем он пытается добавить все, что принадлежит к определенному пользователю:

===================== 
|E-mail | Total | 
|a@a.com | 25  | 
|b@a.com | 8  | 
|c@a.com | 2  | 
===================== 

Я расколоть CSV и добавил значения, что мне нужно в наборе, но я не могу думать о способ добавить нужные мне значения. Есть идеи?

Edit:

Это то, что мой CSV выглядит следующим образом:

p_number,duration,clnup# 
5436715524,00:02:26,2 
6447654246,00:17:18,5 
5996312484,00:01:19,1 
5436715524,00:10:12,6 

Я хотел бы получить общую продолжительность и общий clnup # для каждого уникального p_number. Я сожалею о путанице, но приведенная выше таблица была просто примером.

Answer 1

import csv 

ifile = open('sample.csv', 'rb') 
csv_reader = csv.reader(ifile) 

d = {} 
for row in csv_reader: 
    d[row[0]] = int(row[1]) if d.get(row[0], None) is None else d[row[0]] + int(row[1]) 
from pprint import pprint 
pprint(d) 

Answer 2

Вы можете использовать OrderedDict хранящие имена в качестве значений и обновляют счетчик, как вы идете:

import csv 
from collections import OrderedDict 

od = OrderedDict() 

with open("test.txt") as f: 
    r = csv.reader(f) 
    head = next(r) 
    for name,val in r: 
     od.setdefault(name, 0) 
     od[name] += int(val) 

print(od) 
OrderedDict([('a@a.com', 25), ('b@a.com', 8), ('c@a.com', 2)]) 

Чтобы обновить исходный файл можно записать в NamedTemporaryFile затем использовать shutil.move заменить оригинал после того, как вы написали строки с writerows использованием od.items:

import csv 
from collections import OrderedDict 
from shutil import move 
from tempfile import NamedTemporaryFile 
od = OrderedDict() 

with open("test.txt") as f, NamedTemporaryFile(dir=".",delete=False) as out: 
    r = csv.reader(f) 
    wr = csv.writer(out) 
    head = next(r) 
    wr.writerow(head) 
    for name,val in r: 
     od.setdefault(name, 0) 
     od[name] += int(val) 
    wr.writerows(od.iteritems()) 


move(out.name,"test.txt") 

Выход:

E-mail,M-count 
a@a.com,25 
b@a.com,8 
c@a.com,2 

Если вы не заботитесь о заказе использовать defaultdict вместо:

import csv 

from collections import defaultdict 
from shutil import move 
from tempfile import NamedTemporaryFile 
od = defaultdict(int) 

with open("test.txt") as f, NamedTemporaryFile(dir=".",delete=False) as out: 
    r = csv.reader(f) 
    wr = csv.writer(out) 
    head = next(r) 
    wr.writerow(head) 
    for name,val in r: 
     od[name] += int(val) 
    wr.writerows(od.iteritems())