Plot objective legend (#23)

* Objective plot legend
N-Wouda · Feb 26, 2020 · 2bb17bc · 2bb17bc
1 parent 01afe65
commit 2bb17bc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/alns/Result.py b/alns/Result.py
@@ -87,6 +87,8 @@ def plot_objectives(self, ax=None, title=None, **kwargs):
         ax.set_ylabel("Objective value")
         ax.set_xlabel("Iteration (#)")
 
+        ax.legend(["Current", "Best"], loc="upper right")
+
         plt.draw_if_interactive()
 
     def plot_operator_counts(self, figure=None, title=None, legend=None,

diff --git a/alns/tests/test_result.py b/alns/tests/test_result.py
@@ -67,6 +67,8 @@ def get_objective_plot(ax, data, **kwargs):
     ax.set_ylabel("Objective value")
     ax.set_xlabel("Iteration (#)")
 
+    ax.legend(["Current", "Best"], loc="upper right")
+
 
 def get_operator_plot(figure, destroy, repair, legend=None, suptitle=None,
                       **kwargs):