# HG changeset patch
# User Jordi Gutiérrez Hermoso <jordigh@octave.org>
# Date 1463452173 14400
# Node ID bb6e5cf6aa83f5ff3fdb289ec4226a58108f118a
# Parent  874eb082366004b69d8c26dd51842ddac67d4e56
plots: add geometric data

diff --git a/talk/code/plots.py b/talk/code/plots.py
--- a/talk/code/plots.py
+++ b/talk/code/plots.py
@@ -77,6 +77,7 @@
 girlrect = [0.05, 0.1, 0.9, 0.35]
 girlcolour = 'pink'
 
+fig = plt.figure(figsize=(12,8))
 boxhistplot(men, fig=fig, rect=boyrect, xticks=xticks, colour=boycolour,
             bins=bins, title="Ages of actors")
 boxhistplot(women, fig=fig, rect=girlrect, xticks=xticks, colour=girlcolour,
@@ -89,3 +90,27 @@
 boxhistplot(women, fig=fig, rect=girlrect, xticks=xticks, colour=girlcolour,
             bins=bins, title="Ages of actresses", adjusted=True)
 plt.savefig("boys-and-girls-adjusted.pdf")
+
+np.random.seed(0)
+geo = np.random.geometric(0.20, size=10000)
+
+plt.close('all')
+plt.plot(geo, [1]*geo.size, 'x')
+plt.yticks([])
+plt.savefig("geometric-points.pdf")
+
+plt.close('all')
+plt.boxplot(geo, vert=False, flierprops=dict(marker='x'))
+plt.savefig("geometric-boxplot.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)",
+            xticks=[0, 10, 20, 30, 40, 50])
+plt.savefig("geometric-boxhistplot.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)",
+            xticks=[0, 10, 20, 30, 40, 50], adjusted=True)
+plt.savefig("geometric-boxhistplot-adjusted.pdf")