replace axisgb by facecolor

_unittests/ut_graph/test_corrplot.py

@@ -68,7 +68,7 @@ def test_graph_corrplot(self):
         df = df.corr()
 
         fig = plt.figure(num=None, facecolor='white')
-        ax = plt.subplot(1, 1, 1, aspect='equal', axisbg='white')
+        ax = plt.subplot(1, 1, 1, aspect='equal', facecolor='white')
 
         c = Corrplot(df)
         ax = c.plot(fig=fig, ax=ax)

src/pyensae/graph_helper/corrplot.py

@@ -9,12 +9,20 @@
 :author: Thomas Cokelaer
 :references: http://cran.r-project.org/web/packages/corrplot/vignettes/corrplot-intro.html
 """
-from .linkage import Linkage
+try:
+    from scipy.cluster.hierarchy import dendrogram, fcluster
+except ImportError as e:
+    # The import of scipy was moved it. It creates a warning otherwise.
+    # lib\importlib\_bootstrap.py:205: ImportWarning: can't resolve package
+    # from __spec__ or __package__, falling back on __name__ and __path__
+    import warnings
+    warnings.warn("corrplot.py requires scipy.")
 import numpy as np
 import matplotlib.pyplot as plt
 from matplotlib.patches import Ellipse, Circle, Rectangle, Wedge
 from matplotlib.collections import PatchCollection
 import pandas as pd
+from .linkage import Linkage
 
 
 class Corrplot(Linkage):
@@ -87,7 +95,6 @@ def __init__(self, data, na=0):
             compute_correlation = True
 
         if compute_correlation:
-            print("Computing correlation")
             cor = self.df.corr()
             self.df = cor
 
@@ -116,7 +123,6 @@ def order(self, method='complete', metric='euclidean', inplace=False):
         You probably do not need to use that method. Use :meth:`plot` and
         the two parameters order_metric and order_method instead.
         """
-        from scipy.cluster.hierarchy import fcluster, dendrogram
         Y = self.linkage(self.df, method=method, metric=metric)
         ind1 = fcluster(Y, 0.7 * max(Y[:, 2]), 'distance')
         Z = dendrogram(Y, no_plot=True)
@@ -192,7 +198,6 @@ def plot(self, fig=None, grid=True,
                 else:
                     self.cm = self.cmap_builder(*cmap)
             except Exception:
-                print("incorrect cmap. Use default one")
                 self._set_default_cmap()
         else:
             self._set_default_cmap()
@@ -217,7 +222,7 @@ def plot(self, fig=None, grid=True,
 
         # do we have an axes to plot the data in ?
         if ax is None:
-            ax = plt.subplot(1, 1, 1, aspect='equal', axisbg=axisbg)
+            ax = plt.subplot(1, 1, 1, aspect='equal', facecolor=axisbg)
         else:
             # if so, clear the axes. Colorbar cannot be removed easily.
             plt.sca(ax)