changeset 57:e2d6c57dc1f4

plots.py: add Gaussian data
author Jordi Gutiérrez Hermoso <jordigh@octave.org>
date Tue, 17 May 2016 08:55:45 -0400
parents 7c6fc6af614f
children 57a0f3beaba8
files talk/code/plots.py
diffstat 1 files changed, 33 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
--- a/talk/code/plots.py
+++ b/talk/code/plots.py
@@ -67,6 +67,7 @@
 with open("../../data/women") as f:
     women = [float(x) for x in f.readlines()]
 
+# curiousgnu's data
 
 xticks = np.arange(5,105,5)
 bins = 0.5 + np.arange(0,100)
@@ -91,6 +92,8 @@
             bins=bins, title="Ages of actresses", adjusted=True)
 plt.savefig("boys-and-girls-adjusted.pdf")
 
+# Geometric distro
+
 np.random.seed(0)
 geo = np.random.geometric(0.20, size=10000)
 
@@ -103,6 +106,9 @@
 plt.boxplot(geo, vert=False, flierprops=dict(marker='x'))
 plt.savefig("geometric-boxplot.pdf")
 
+plt.axis('off')
+plt.savefig("geometric-boxplot-bare.pdf")
+
 fig = plt.figure(figsize=(12,8))
 boxhistplot(geo, fig=fig, rect=[0.05, 0.1, 0.9, 0.7], colour='yellow',
             bins=20, title="MS lesion counts (simulated)",
@@ -114,3 +120,30 @@
             bins=20, title="MS lesion counts (simulated)",
             xticks=[0, 10, 20, 30, 40, 50], adjusted=True)
 plt.savefig("geometric-boxhistplot-adjusted.pdf")
+
+# Normal distro
+N = np.random.normal(size=1000)
+
+plt.close('all')
+plt.plot(N, [1]*N.size, 'x')
+plt.yticks([])
+plt.savefig("normal-points.pdf")
+
+plt.close('all')
+plt.boxplot(N, vert=False, flierprops=dict(marker='x'))
+plt.savefig("normal-boxplot.pdf")
+
+plt.axis('off')
+plt.savefig("normal-boxplot-bare.pdf")
+
+fig = plt.figure(figsize=(12,8))
+boxhistplot(N, fig=fig, rect=[0.05, 0.1, 0.9, 0.7], colour='green',
+            bins=20, title="Brain volumes (simulated)",
+            xticks=range(-4,5))
+plt.savefig("normal-boxhistplot.pdf")
+
+fig = plt.figure(figsize=(12,8))
+boxhistplot(N, fig=fig, rect=[0.05, 0.1, 0.9, 0.7], colour='green',
+            bins=20, title="Brain volumes (simulated)",
+            xticks=range(-4,5), adjusted=True)
+plt.savefig("normal-boxhistplot-adjusted.pdf")