support axes in plot_decision_regions

rasbt · May 7, 2017 · 1ddaea8 · 1ddaea8
1 parent 0d1643c
commit 1ddaea8
Show file tree

Hide file tree

Showing 11 changed files with 72 additions and 26 deletions.
diff --git a/docs/sources/CHANGELOG.md b/docs/sources/CHANGELOG.md
@@ -25,6 +25,7 @@ The CHANGELOG for the current development version is available at
 ##### Bug Fixes
 
 - `SequentialFeatureSelector` now correctly accepts a `None` argument for the `scoring` parameter to infer the default scoring metric from scikit-learn classifiers and regressors.
+- The `plot_decision_regions` function now supports pre-existing axes objects generated via matplotlib's `plt.subplots`. ([See example](http://rasbt.github.io/mlxtend/user_guide/plotting/plot_decision_regions/#example-6-working-with-existing-axes-objects-using-subplots))
 
 
 ### Version 0.6.0 (2017-03-18)

diff --git a/docs/sources/user_guide/plotting/plot_decision_regions.ipynb b/docs/sources/user_guide/plotting/plot_decision_regions.ipynb
diff --git a/.../user_guide/plotting/plot_decision_regions_files/plot_decision_regions_11_0.png b/.../user_guide/plotting/plot_decision_regions_files/plot_decision_regions_11_0.png
diff --git a/.../user_guide/plotting/plot_decision_regions_files/plot_decision_regions_13_0.png b/.../user_guide/plotting/plot_decision_regions_files/plot_decision_regions_13_0.png
diff --git a/.../user_guide/plotting/plot_decision_regions_files/plot_decision_regions_19_0.png b/.../user_guide/plotting/plot_decision_regions_files/plot_decision_regions_19_0.png
diff --git a/.../user_guide/plotting/plot_decision_regions_files/plot_decision_regions_21_0.png b/.../user_guide/plotting/plot_decision_regions_files/plot_decision_regions_21_0.png
diff --git a/.../user_guide/plotting/plot_decision_regions_files/plot_decision_regions_23_0.png b/.../user_guide/plotting/plot_decision_regions_files/plot_decision_regions_23_0.png
diff --git a/.../user_guide/plotting/plot_decision_regions_files/plot_decision_regions_25_0.png b/.../user_guide/plotting/plot_decision_regions_files/plot_decision_regions_25_0.png
diff --git a/...s/user_guide/plotting/plot_decision_regions_files/plot_decision_regions_6_0.png b/...s/user_guide/plotting/plot_decision_regions_files/plot_decision_regions_6_0.png
diff --git a/...s/user_guide/plotting/plot_decision_regions_files/plot_decision_regions_8_0.png b/...s/user_guide/plotting/plot_decision_regions_files/plot_decision_regions_8_0.png
diff --git a/mlxtend/plotting/decision_regions.py b/mlxtend/plotting/decision_regions.py
@@ -9,7 +9,6 @@
 from itertools import cycle
 import matplotlib.pyplot as plt
 import numpy as np
-from matplotlib.colors import ListedColormap
 
 
 def plot_decision_regions(X, y, clf,
@@ -147,14 +146,8 @@ def plot_decision_regions(X, y, clf,
         ax.axes.get_yaxis().set_ticks([])
 
     if legend:
-        legend = plt.legend(loc=legend,
-                            fancybox=True,
-                            framealpha=0.3,
-                            scatterpoints=1,
-                            handletextpad=-0.25,
-                            borderaxespad=0.9)
-
-        ax.add_artist(legend)
+        handles, labels = ax.get_legend_handles_labels()
+        ax.legend(handles, labels, framealpha=0.3, scatterpoints=1, loc=legend)
 
     if plot_testdata:
         if dim == '2d':