summaryrefslogtreecommitdiff
path: root/average.py
blob: 01a98dc12d11a1a8409d08f64634c415b92a20c7 (about) (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
from collections import defaultdict
from itertools import imap as map
from operator import itemgetter

RESULTS_FILE = 'results.csv'
results = defaultdict(list)

for line in open(RESULTS_FILE, 'rU'):
    game, name, length, time = line[:-1].split(',')
    length = int(length)
    time = int(time)
    results[name].append((length, time))

def average(items):
    s = 0.0
    l = 0
    for item in items:
        s += item
        l += 1
    return s / l

lengths = []
times = []

for name, series in results.items():
    length_average = average(map(itemgetter(0), series))
    lengths.append((length_average, name))
    time_average = average(map(itemgetter(1), series))
    times.append((time_average, name))

lengths.sort(reverse=True)
times.sort(reverse=True)

print 'Lengths'
print '======='
for length, name in lengths:
    print name, length
print

print 'Times'
print '====='
for time, name in times:
    print name, time