summaryrefslogtreecommitdiffstats
path: root/hawkes/plot.py
blob: 24d3f352b6241a4f6cc29453870b66df53f46a90 (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
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()