changeset 54:bb6e5cf6aa83

plots: add geometric data
author Jordi Gutiérrez Hermoso <jordigh@octave.org>
date Mon, 16 May 2016 22:29:33 -0400
parents 874eb0823660
children 0f0b16b3c6f5
files talk/code/plots.py
diffstat 1 files changed, 25 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
--- 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")