added some snippets

rose/__init__.py

@@ -84,7 +84,7 @@ def bar(self, **kwargs):
 
         return ax
 
-    def plot(self, spacing=3, figsize=(7, 6), colors="viridis", **kwargs):
+    def plot(self, ax=None, spacing=3, figsize=(7, 6), colors="viridis", **kwargs):
         """Plot the binned data as a windrose
 
         Parameters
@@ -121,8 +121,10 @@ def plot(self, spacing=3, figsize=(7, 6), colors="viridis", **kwargs):
         nsectors, nbins = table.shape
         radian_locs = np.arange(0, np.pi * 2, np.pi * 2 / len(table))
 
-        fig = plt.figure(figsize=figsize)
-        ax = plt.subplot(projection="polar")
+        if ax is None:
+            fig = plt.figure(figsize=figsize)
+            ax = plt.subplot(projection="polar")
+
         ax.set_theta_zero_location("N")  # theta=0 at the top
         ax.set_theta_direction(-1)  # theta increasing clockwise
 

rose/snippets.py

@@ -0,0 +1,23 @@
+"""
+Some helpful snippets for pandas-rose.
+"""
+import numpy as np
+import matplotlib.pyplot as plt
+
+def labels_on_top(ax):
+    """Put radius labels on top of the bar plot."""
+    for i in ax.get_yticks()[::2]:
+        # TODO: determine if values on ax are normed
+        #if table_kwargs.get("normed", None):
+        #    text = f"{i:.1%}"
+        #else:
+        #    text = f"{i:g}"
+        text = i
+        ax.text(np.pi / 16, i, text, zorder=100, ha="center")
+    ax.set_yticklabels([])
+    return ax
+
+def generate_polar_subplots(nrows, ncols):
+    """Generate a matrix of polar subplots"""
+    return plt.subplots(nrows, ncols, subplot_kw=dict(projection="polar"))
+