delay pyplot import

Author: TomAugspurger

pandas/plotting/_matplotlib/boxplot.py

@@ -1,7 +1,6 @@
 from collections import namedtuple
 import warnings
 
-from matplotlib import pyplot as plt
 from matplotlib.artist import setp
 import numpy as np
 
@@ -216,6 +215,7 @@ def boxplot(data, column=None, by=None, ax=None, fontsize=None,
             rot=0, grid=True, figsize=None, layout=None, return_type=None,
             **kwds):
 
+    import matplotlib.pyplot as plt
     # validate return_type:
     if return_type not in BoxPlot._valid_return_types:
         raise ValueError("return_type must be {'axes', 'dict', 'both'}")
@@ -297,6 +297,7 @@ def plot_group(keys, values, ax):
 def boxplot_frame(self, column=None, by=None, ax=None, fontsize=None, rot=0,
                   grid=True, figsize=None, layout=None,
                   return_type=None, **kwds):
+    import matplotlib.pyplot as plt
     converter._WARN = False  # no warning for pandas plots
     ax = boxplot(self, column=column, by=by, ax=ax, fontsize=fontsize,
                  grid=grid, rot=rot, figsize=figsize, layout=layout,

pandas/plotting/_matplotlib/core.py

@@ -2,7 +2,6 @@
 from typing import Optional  # noqa
 import warnings
 
-import matplotlib.pyplot as plt
 import numpy as np
 
 from pandas._config import get_option
@@ -61,6 +60,7 @@ def __init__(self, data, kind=None, by=None, subplots=False, sharex=None,
                  secondary_y=False, colormap=None,
                  table=False, layout=None, **kwds):
 
+        import matplotlib.pyplot as plt
         converter._WARN = False  # no warning for pandas plots
         self.data = data
         self.by = by
@@ -104,7 +104,7 @@ def __init__(self, data, kind=None, by=None, subplots=False, sharex=None,
             self.rot = self._default_rot
 
         if grid is None:
-            grid = False if secondary_y else self.plt.rcParams['axes.grid']
+            grid = False if secondary_y else plt.rcParams['axes.grid']
 
         self.grid = grid
         self.legend = legend
@@ -619,6 +619,8 @@ def _get_ax(self, i):
 
     @classmethod
     def get_default_ax(cls, ax):
+        import matplotlib.pyplot as plt
+
         if ax is None and len(plt.get_fignums()) > 0:
             with plt.rc_context():
                 ax = plt.gca()

pandas/plotting/_matplotlib/hist.py

@@ -1,6 +1,5 @@
 import warnings
 
-import matplotlib.pyplot as plt
 import numpy as np
 
 from pandas.core.dtypes.common import is_integer, is_list_like
@@ -222,6 +221,7 @@ def plot_group(group, ax):
 def hist_series(self, by=None, ax=None, grid=True, xlabelsize=None,
                 xrot=None, ylabelsize=None, yrot=None, figsize=None,
                 bins=10, **kwds):
+    import matplotlib.pyplot as plt
     if by is None:
         if kwds.get('layout', None) is not None:
             raise ValueError("The 'layout' keyword is not supported when "

pandas/plotting/_matplotlib/misc.py

@@ -2,7 +2,6 @@
 
 import matplotlib.lines as mlines
 import matplotlib.patches as patches
-import matplotlib.pyplot as plt
 import numpy as np
 
 from pandas.core.dtypes.missing import notna
@@ -105,6 +104,7 @@ def _get_marker_compat(marker):
 
 
 def radviz(frame, class_column, ax=None, color=None, colormap=None, **kwds):
+    import matplotlib.pyplot as plt
 
     def normalize(series):
         a = min(series)
@@ -169,6 +169,7 @@ def normalize(series):
 
 def andrews_curves(frame, class_column, ax=None, samples=200, color=None,
                    colormap=None, **kwds):
+    import matplotlib.pyplot as plt
 
     def function(amplitudes):
         def f(t):
@@ -224,6 +225,7 @@ def f(t):
 
 def bootstrap_plot(series, fig=None, size=50, samples=500, **kwds):
 
+    import matplotlib.pyplot as plt
     # random.sample(ndarray, int) fails on python 3.3, sigh
     data = list(series.values)
     samplings = [random.sample(data, size) for _ in range(samples)]
@@ -270,6 +272,7 @@ def parallel_coordinates(frame, class_column, cols=None, ax=None, color=None,
                          use_columns=False, xticks=None, colormap=None,
                          axvlines=True, axvlines_kwds=None, sort_labels=False,
                          **kwds):
+    import matplotlib.pyplot as plt
     if axvlines_kwds is None:
         axvlines_kwds = {'linewidth': 1, 'color': 'black'}
 
@@ -336,6 +339,7 @@ def parallel_coordinates(frame, class_column, cols=None, ax=None, color=None,
 
 def lag_plot(series, lag=1, ax=None, **kwds):
     # workaround because `c='b'` is hardcoded in matplotlibs scatter method
+    import matplotlib.pyplot as plt
     kwds.setdefault('c', plt.rcParams['patch.facecolor'])
 
     data = series.values
@@ -350,6 +354,8 @@ def lag_plot(series, lag=1, ax=None, **kwds):
 
 
 def autocorrelation_plot(series, ax=None, **kwds):
+    import matplotlib.pyplot as plt
+
     n = len(series)
     data = np.asarray(series)
     if ax is None:

pandas/plotting/_matplotlib/style.py

@@ -3,7 +3,6 @@
 
 import matplotlib.cm as cm
 import matplotlib.colors
-import matplotlib.pyplot as plt
 import numpy as np
 
 from pandas.core.dtypes.common import is_list_like
@@ -13,6 +12,7 @@
 
 def _get_standard_colors(num_colors=None, colormap=None, color_type='default',
                          color=None):
+    import matplotlib.pyplot as plt
     if color is None and colormap is not None:
         if isinstance(colormap, str):
             cmap = colormap

pandas/plotting/_matplotlib/timeseries.py

@@ -3,8 +3,6 @@
 import functools
 import warnings
 
-from matplotlib import pylab
-import matplotlib.pyplot as plt
 import numpy as np
 
 from pandas._libs.tslibs.frequencies import (
@@ -42,6 +40,7 @@ def tsplot(series, plotf, ax=None, **kwargs):
     .. deprecated:: 0.23.0
        Use Series.plot() instead
     """
+    import matplotlib.pyplot as plt
     warnings.warn("'tsplot' is deprecated and will be removed in a "
                   "future version. Please use Series.plot() instead.",
                   FutureWarning, stacklevel=2)
@@ -323,6 +322,7 @@ def format_dateaxis(subplot, freq, index):
     default, changing the limits of the x axis will intelligently change
     the positions of the ticks.
     """
+    from matplotlib import pylab
 
     # handle index specific formatting
     # Note: DatetimeIndex does not use this

pandas/plotting/_matplotlib/tools.py

@@ -2,7 +2,6 @@
 from math import ceil
 import warnings
 
-import matplotlib.pyplot as plt
 import matplotlib.table
 import matplotlib.ticker as ticker
 import numpy as np
@@ -168,6 +167,7 @@ def _subplots(naxes=None, sharex=False, sharey=False, squeeze=True,
     # Four polar axes
     plt.subplots(2, 2, subplot_kw=dict(polar=True))
     """
+    import matplotlib.pyplot as plt
     if subplot_kw is None:
         subplot_kw = {}
 
@@ -345,6 +345,7 @@ def _get_xlim(lines):
 
 def _set_ticks_props(axes, xlabelsize=None, xrot=None,
                      ylabelsize=None, yrot=None):
+    import matplotlib.pyplot as plt
     for ax in _flatten(axes):
         if xlabelsize is not None:
             plt.setp(ax.get_xticklabels(), fontsize=xlabelsize)