summaryrefslogtreecommitdiffstats
path: root/twitter/dist.py
blob: 8902869d02f9773a33908e7355db1049eaaa0e07 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
import sys
import os.path


def compute_dist(filename):
    f = open(filename)
    users = {}
    for line in f:
        values = line.strip().split()
        users[values[0]] = int(values[2])
    print sum(users.itervalues()) / float(len(users))

    of = open(os.path.splitext(filename)[0] + "_dist.txt", "w")
    for degree in users.itervalues():
        of.write(str(degree) + "\n")

if __name__ == "__main__":
    compute_dist(sys.argv[1])