From 3d03e91e78f0ffbb49e7842308ed70946c7a6c2f Mon Sep 17 00:00:00 2001 From: Joris Van den Bossche Date: Mon, 30 Sep 2013 18:39:14 +0200 Subject: [PATCH] VIS: let scatter plots obey mpl color scheme (#3338) --- pandas/tools/plotting.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/pandas/tools/plotting.py b/pandas/tools/plotting.py index 6631a3cf8c6f1..d6c0482d86be4 100644 --- a/pandas/tools/plotting.py +++ b/pandas/tools/plotting.py @@ -229,6 +229,7 @@ def scatter_matrix(frame, alpha=0.5, figsize=None, ax=None, grid=False, >>> df = DataFrame(np.random.randn(1000, 4), columns=['A','B','C','D']) >>> scatter_matrix(df, alpha=0.2) """ + import matplotlib.pyplot as plt from matplotlib.artist import setp df = frame._get_numeric_data() @@ -246,6 +247,9 @@ def scatter_matrix(frame, alpha=0.5, figsize=None, ax=None, grid=False, hist_kwds = hist_kwds or {} density_kwds = density_kwds or {} + # workaround because `c='b'` is hardcoded in matplotlibs scatter method + kwds.setdefault('c', plt.rcParams['patch.facecolor']) + for i, a in zip(lrange(n), df.columns): for j, b in zip(lrange(n), df.columns): ax = axes[i, j] @@ -653,6 +657,10 @@ def lag_plot(series, lag=1, ax=None, **kwds): ax: Matplotlib axis object """ import matplotlib.pyplot as plt + + # workaround because `c='b'` is hardcoded in matplotlibs scatter method + kwds.setdefault('c', plt.rcParams['patch.facecolor']) + data = series.values y1 = data[:-lag] y2 = data[lag:] @@ -1889,6 +1897,9 @@ def scatter_plot(data, x, y, by=None, ax=None, figsize=None, grid=False, **kwarg """ import matplotlib.pyplot as plt + # workaround because `c='b'` is hardcoded in matplotlibs scatter method + kwargs.setdefault('c', plt.rcParams['patch.facecolor']) + def plot_group(group, ax): xvals = group[x].values yvals = group[y].values