python - 哈希表中的 CSV 然后计算总和

Question

这是我的 csv 文件:

name;value
John;4.0
John;-15.0
John;1.0
John;-2.0
Bob;1.0
Bob;2.5
Bob;-8

我想打印这个输出:

John : 22
Bob : 11,5

22 因为 4+15+1+2=22

11,5 因为 1+2,5+8 = 11,5

重要的是忽略-符号并用正号计算总数。

我试过这个:

import csv
with open('myfile.csv', 'rb') as f:
    reader = csv.reader(f, delimiter=';')
    for row in reader:
        print row
Hashtable = {}

我知道我必须使用具有键值系统的哈希表，但我卡在了这一点上，请帮助我，我正在使用 python 2.7。

Answer 1

假设11,5应该是11.5，使用defaultdict来处理重复键，只是str.lstrip > 任何减号和+=每个值

import csv
from collections import defaultdict

d = defaultdict(float)
with open('test.txt', 'rb') as f:
    reader = csv.reader(f, delimiter=';')
    next(reader) # skip header
    for name, val in reader:
        d[name] += float(val.lstrip("-"))

输出:

for k,v in d.items():
    print(k,v)
('Bob', 11.5)
('John', 22.0)

如果出于某种原因你想使用普通的字典，你可以使用dict.setdefault:

d = {}
with open('test.txt', 'rb') as f:
    reader = csv.reader(f, delimiter=';')
    next(reader)
    for name, val in reader:
        d.setdefault(name, 0)
        d[name] += float(val.lstrip("-"))

使用 defauldict 和 lstrip 是最有效的，一些时机:

In [26]: timeit default()
100 loops, best of 3: 2.6 ms per loop

In [27]: timeit counter()
100 loops, best of 3: 3.98 ms per loop