1 files changed, 18 insertions, 0 deletions
diff --git a/twitter/dist.py b/twitter/dist.py
new file mode 100644
index 0000000..8902869
--- /dev/null
+++ b/twitter/dist.py
@@ -0,0 +1,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])