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

diff --git a/wikibios/scatterplot.py b/wikibios/scatterplot.py
new file mode 100644
index 0000000..61582d8
--- /dev/null
+++ b/wikibios/scatterplot.py
@@ -0,0 +1,21 @@
+"""Make a scatterplot of male and female articles."""
+import wikibios
+from datetime import datetime
+from matplotlib import pyplot
+
+figure = pyplot.figure()
+axes = figure.gca()
+
+firstedits_male = wikibios.columns_male['firstedit']
+birth_years_male = wikibios.columns_male['birth_year']
+axes.scatter(firstedits_male, birth_years_male, alpha=0.1, c='green', edgecolors='None', label='Male')
+
+firstedits_female = wikibios.columns_female['firstedit']
+birth_years_female = wikibios.columns_female['birth_year']
+axes.scatter(firstedits_female, birth_years_female, alpha=0.1, c='orange', edgecolors='None', label='Female')
+axes.set_xlabel('Article Date')
+axes.set_ylabel('Birth Year')
+axes.legend()
+
+figure.savefig('scatter.png')
+