From 0b0d26b2c0ed7d6f0b69f550f6584dabae184ba1 Mon Sep 17 00:00:00 2001
From: Thibaut Horel <thibaut.horel@gmail.com>
Date: Sun, 23 Aug 2015 03:47:58 -0700
Subject: Code for hawkes model

---
 hawkes/plot.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)
 create mode 100644 hawkes/plot.py

(limited to 'hawkes/plot.py')

diff --git a/hawkes/plot.py b/hawkes/plot.py
new file mode 100644
index 0000000..24d3f35
--- /dev/null
+++ b/hawkes/plot.py
@@ -0,0 +1,26 @@
+from mpl_toolkits.mplot3d import Axes3D
+from matplotlib import cm
+import matplotlib.pyplot as plt
+
+fig = plt.figure()
+ax = fig.gca(projection='3d')
+d = {}
+for line in open("values.txt"):
+    v = map(float, line.strip().split())
+    try:
+        if (v[0], v[2]) in d:
+            d[(v[0], v[2])] = min(d[(v[0], v[2])], v[3])
+        else:
+            d[(v[0], v[2])] = v[3]
+    except:
+        continue
+
+x, y, z = [], [], []
+for k, v in d.iteritems():
+    x.append(k[0] / 1000000.)
+    y.append(k[1] / 1000000.)
+    z.append(v / 100000.)
+
+print x
+surf = ax.plot_trisurf(x, y, z)
+plt.show()
-- 
cgit v1.2.3-70-g09d2