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

[matplotlib-cdsw] / wikibios / fractionplot.py

diff --git a/wikibios/fractionplot.py b/wikibios/fractionplot.py
new file mode 100644 (file)
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')