1 files changed, 30 insertions, 0 deletions
diff --git a/distribution.py b/distribution.py
new file mode 100644
index 0000000..49d2d5e
--- /dev/null
+++ b/distribution.py
@@ -0,0 +1,30 @@
+import matplotlib.pyplot as plt
+import numpy as np
+import sys
+
+
+def load_distribution(filename):
+    l = [int(line.strip()) for line in open(filename)]
+    l = sorted(l)
+    l = l[:]
+    return l
+
+
+def plot_distribution(files):
+    for file in files:
+        x = load_distribution(file)
+        a = np.array(x)
+        print file, a.mean(), a.size
+        n, bins, patches = plt.hist(x, 50, normed=1, facecolor='green')
+        plt.show()
+        # # n_nodes = float(sum(y))
+        # plt.plot(x, np.array(y), ".")
+        # if save:
+        #     fig = plt.gcf()
+        #     fig.set_size_inches(20, 15)
+        #     # plt.savefig(output, bbox_inches="tight")
+        # else:
+        #     plt.show()
+
+if __name__ == "__main__":
+    plot_distribution(sys.argv[1:])