merged in all the code from ariel's branches and moved wikibios to a subdir

[matplotlib-cdsw] / wikibios / histograms.py

diff --git a/wikibios/histograms.py b/wikibios/histograms.py
new file mode 100644 (file)
index 0000000..e9a3705
--- /dev/null
+++ b/wikibios/histograms.py
@@ -0,0 +1,23 @@
+"""Make combined histograms of the male and female articles, by birth year and first edit date."""
+import wikibios
+from matplotlib import pyplot, dates
+
+figure = pyplot.figure()
+
+axes1 = figure.add_subplot(2, 1, 1)
+birth_years_male = wikibios.columns_male['birth_year']
+birth_years_female =  wikibios.columns_female['birth_year']
+axes1.hist([birth_years_female, birth_years_male], bins=50, stacked=True, range=[1514, 2014])
+axes1.set_xlabel('Birth Year')
+axes1.set_ylabel('Articles')
+
+axes2 = figure.add_subplot(2, 1, 2)
+firstedits_male = dates.date2num(wikibios.columns_male['firstedit'])
+firstedits_female = dates.date2num(wikibios.columns_female['firstedit'])
+axes2.hist([firstedits_female, firstedits_male], bins=50, stacked=True, label=['Female', 'Male'])
+axes2.xaxis.set_major_formatter(dates.AutoDateFormatter(dates.AutoDateLocator()))
+axes2.legend()
+axes2.set_xlabel('Article Year')
+axes2.set_ylabel('Articles')
+
+figure.savefig('histograms.pdf')