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

diff --git a/wikibios/fractionplot.py b/wikibios/fractionplot.py
new file mode 100644
index 0000000..eb719a3
--- /dev/null
+++ b/wikibios/fractionplot.py
@@ -0,0 +1,53 @@
+"""Plot the fraction of articles that are female, by birth year and first edit date."""
+import wikibios
+from matplotlib import pyplot
+from operator import itemgetter
+
+figure = pyplot.figure()
+
+rows_by_birth_year = sorted(wikibios.rows, key=itemgetter('birth_year'))
+birth_year_medians = []
+fraction_female_by_birth_year = []
+N = 1000
+i = 0
+while i + N <= len(rows_by_birth_year):
+	chunk = rows_by_birth_year[i:i+N]
+	i = i + N
+
+	birth_year_medians.append(chunk[N / 2]['birth_year'])
+
+	count_female = 0.0
+	for row in chunk:
+		if row['gender'] == 'female':
+			count_female = count_female + 1
+	fraction_female = count_female / N
+	fraction_female_by_birth_year.append(fraction_female)
+
+axes1 = figure.add_subplot(2, 1, 1)
+axes1.plot(birth_year_medians, fraction_female_by_birth_year)
+axes1.set_xlabel('Birth Year')
+axes1.set_ylabel('Fraction Female')
+
+rows_by_firstedit = sorted(wikibios.rows, key=itemgetter('firstedit'))
+fraction_female_by_firstedit = []
+firstedit_medians = []
+i = 0
+while i + N <= len(rows_by_birth_year):
+	chunk = rows_by_firstedit[i:i+N]
+	i = i + N
+
+	firstedit_medians.append(chunk[N / 2]['firstedit'])
+
+	count_female = 0.0
+	for row in chunk:
+		if row['gender'] == 'female':
+			count_female = count_female + 1
+	fraction_female = count_female / N
+	fraction_female_by_firstedit.append(fraction_female)
+
+axes2 = figure.add_subplot(2, 1, 2)
+axes2.plot(firstedit_medians, fraction_female_by_firstedit)
+axes2.set_xlabel('Article Year')
+axes2.set_ylabel('Fraction Female')
+
+figure.savefig('fractions.pdf')