Deathlog/plotter.py at master · ChefCarlos/Deathlog · GitHub

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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
import matplotlib.pyplot as plt
import numpy as np
from   scipy.stats import multivariate_normal

means = [[0.69435315 0.78959935]
 [0.40464967 0.80077265],
 [0.61103521 0.49641971],
 [0.85978445 0.6656501 ],
 [0.26533757 0.87580696],
 [0.5733805  0.67776969],
 [0.71125827 0.65690596],
 [0.88267917 0.78811334],
 [0.5407539  0.47052626],
 [0.68961372 0.43372022],
 [0.3888409  0.65202992],
 [0.78942785 0.45272085],
 [0.52199914 0.66200502],
 [0.48263736 0.35893212],
 [0.6260637  0.57366876],
 [0.79133867 0.59013982],
 [0.78196506 0.80864053],
 [0.34788716 0.86058025],
 [0.45460949 0.78467196],
 [0.42512422 0.86627887]]

variances = [[[ 2.13813430e-04  4.71236211e-05]
        [ 4.71236211e-05  4.20364402e-04]]
 [[ 7.74188209e-05 -5.37348572e-05]
  [-5.37348572e-05  2.01561069e-04]]
 [[ 2.50796004e-05  3.29168679e-05]
  [ 3.29168679e-05  5.36694830e-04]]
 [[ 2.98070997e-04  3.88354294e-04]
  [ 3.88354294e-04  1.67939601e-03]]
 [[ 2.06474829e-04 -4.20069552e-05]
  [-4.20069552e-05  1.87885296e-03]]
 [[ 7.41280049e-04 -3.87979302e-04]
  [-3.87979302e-04  1.93853233e-03]]
 [[ 7.85151052e-04  3.38001190e-04]
  [ 3.38001190e-04  7.29185610e-04]]
 [[ 6.97750427e-04 -3.97686586e-04]
  [-3.97686586e-04  1.00202404e-03]]
 [[ 3.51387666e-04 -4.09037475e-05]
  [-4.09037475e-05  7.99984406e-04]]

 [[ 3.80734353e-04 -6.30393264e-07]
  [-6.30393264e-07  8.00975480e-04]]

 [[ 4.10696089e-03 -1.20338242e-03]
  [-1.20338242e-03  4.46606915e-03]]

 [[ 2.22843055e-04  3.01681684e-04]
  [ 3.01681684e-04  1.08918168e-03]]

 [[ 5.94594186e-04 -2.40560498e-04]
  [-2.40560498e-04  1.08216986e-03]]

 [[ 1.80531368e-04 -4.48955504e-04]
  [-4.48955504e-04  3.55592979e-03]]

 [[ 2.43842453e-04 -8.16423412e-05]
  [-8.16423412e-05  3.04475101e-04]]

 [[ 1.77311264e-04 -7.66106772e-05]
  [-7.66106772e-05  6.69425605e-04]]

 [[ 6.23955836e-04  1.06532491e-04]
  [ 1.06532491e-04  1.32984786e-03]]

 [[ 1.03112664e-03 -7.45560524e-06]
  [-7.45560524e-06  5.93582441e-04]]

 [[ 1.27094370e-03  1.99511155e-04]
  [ 1.99511155e-04  6.28247174e-04]]

 [[ 3.56479937e-04 -3.56028063e-05]
  [-3.56028063e-05  4.95039206e-04]]]

N    = 200
X    = np.linspace(-4, 4, N)
Y    = np.linspace(-4, 4, N)
X, Y = np.meshgrid(X, Y)
pos  = np.dstack((X, Y))
rv   = multivariate_normal([0, 1], [[1, 0.8], [0.8, 1]])
Z    = rv.pdf(pos)

plt.contour(X, Y, Z)
plt.show()