Merge pull request #143 from timkpaine/scattermat

resolve scatter mat for later pandas
pmorissette · Apr 24, 2021 · 0575a43 · 0575a43
2 parents ebcd62f + 0a0e082
commit 0575a43
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ffn/core.py b/ffn/core.py
@@ -1042,7 +1042,10 @@ def plot_scatter_matrix(self, freq=None, title=None, figsize=(10, 10), **kwargs)
 
         plt.figure()
         ser = self._get_series(freq).to_returns().dropna()
-        pd.scatter_matrix(ser, figsize=figsize, **kwargs)
+        if hasattr(pd, "scatter_matrix"):
+            pd.scatter_matrix(ser, figsize=figsize, **kwargs)
+        else:
+            pd.plotting.scatter_matrix(ser, figsize=figsize, **kwargs)
         return plt.suptitle(title)
 
     def plot_histograms(self, freq=None, title=None, figsize=(10, 10), **kwargs):