Plot for distribution of wins

Author: marcharper

axelrod/plot.py

@@ -1,3 +1,5 @@
+import numpy
+
 matplotlib_installed = True
 try:
     import matplotlib.pyplot as plt
@@ -11,7 +13,6 @@ class Plot(object):
 
     def __init__(self, result_set):
         self.result_set = result_set
-        # self._nplayers = self.result_set.nplayers
         self.matplotlib_installed = matplotlib_installed
 
     @property
@@ -55,6 +56,45 @@ def boxplot(self):
         plt.title(self._boxplot_title)
         return figure
 
+    @property
+    def _winplot_dataset(self):
+        # Sort wins by median
+        wins = self.result_set.wins
+        ranked_names = self.result_set.ranked_names
+        medians = map(numpy.median, wins)
+        players = self.result_set.players
+        medians = sorted([(m, i) for (i, m) in enumerate(medians)], reverse=True)
+        # Reorder and grab names
+        wins = [wins[x[1]] for x in medians]
+        ranked_names = [str(players[x[1]]) for x in medians]
+        #zipped = [(wins[i], str(self.players[i])) for (m, i) in medians]
+        #wins, ranked_names = zip(*zipped) # unzip!
+        return wins, ranked_names
+
+    @property
+    def _winplot_title(self):
+        return ("Distribution of wins"
+                " {} turns repeated {} times ({} strategies)").format(
+            self.result_set.turns,
+            self.result_set.repetitions,
+            len(self.result_set.ranking))
+
+    def winplot(self):
+        if not self.matplotlib_installed:
+            return None
+
+        wins, ranked_names = self._winplot_dataset
+
+        figure = plt.figure()
+        plt.boxplot(wins)
+        plt.xticks(
+            self._boxplot_xticks_locations,
+            ranked_names,
+            rotation=90)
+        plt.tick_params(axis='both', which='both', labelsize=7)
+        plt.title(self._winplot_title)
+        return figure
+
     def payoff(self):
 
         if not self.matplotlib_installed:

axelrod/tournament_manager.py

@@ -126,7 +126,7 @@ def _save_plots(self, tournament, ecosystem=None, image_format="svg"):
             self._logger.error('The matplotlib library is not installed. '
                             'No plots will be produced')
             return
-        for plot_type in ('boxplot', 'payoff'):
+        for plot_type in ('boxplot', 'payoff', 'winplot'):
             figure = getattr(plot, plot_type)()
             file_name = self._output_file_path(
                 tournament.name + '_' + plot_type, image_format)