X-Git-Url: https://projects.mako.cc/source/matplotlib-cdsw/blobdiff_plain/6e94163bfc436e165e143cbc6259054537f554a9..f53e20bf3dc31dc78d0cce4426f74057521f8e68:/wikibios/hist2d.py

diff --git a/wikibios/hist2d.py b/wikibios/hist2d.py
new file mode 100644
index 0000000..119a30a
--- /dev/null
+++ b/wikibios/hist2d.py
@@ -0,0 +1,29 @@
+"""Make a 2D histogram ("heat map") of the male and female articles by birth year and first edit date"""
+import wikibios
+from datetime import datetime
+from matplotlib import pyplot, dates
+
+figure = pyplot.figure()
+
+min_firstedit = min(dates.date2num(wikibios.columns['firstedit']))
+max_firstedit = max(dates.date2num(wikibios.columns['firstedit']))
+
+firstedits_male = dates.date2num(wikibios.columns_male['firstedit'])
+birth_years_male = wikibios.columns_male['birth_year']
+
+axes1 = figure.add_subplot(1,2,1)
+h1 = axes1.hist2d(firstedits_male, birth_years_male, range=[[min_firstedit, max_firstedit], [1750, 2014]], bins=50)
+figure.colorbar(h1[3])
+axes1.xaxis.set_major_formatter(dates.AutoDateFormatter(dates.AutoDateLocator()))
+axes1.set_title('Male')
+
+firstedits_female = dates.date2num(wikibios.columns_female['firstedit'])
+birth_years_female = wikibios.columns_female['birth_year']
+
+axes2 = figure.add_subplot(1,2,2)
+h2 = axes2.hist2d(firstedits_female, birth_years_female, range=[[min_firstedit, max_firstedit], [1750, 2014]], bins=50)
+figure.colorbar(h2[3])
+axes2.xaxis.set_major_formatter(dates.AutoDateFormatter(dates.AutoDateLocator()))
+axes2.set_title('Female')
+
+figure.savefig('hist2d.png')