[common] improved axes argument handling

src/plottools/common.py

@@ -44,8 +44,15 @@ def common_xlabels(fig, *axes):
         axes = fig.get_axes()
     if len(axes) == 0:
         return
-    if len(axes) == 1 and isinstance(axes[0], np.ndarray):
-        axes = axes[0].ravel()
+    axs = []
+    for ax in axes:
+        if isinstance(ax, np.ndarray):
+            axs.extend(ax.ravel())
+        elif isinstance(ax, (tuple, list)):
+            axs.extend(ax)
+        else:
+            axs.append(ax)
+    axes = axs
     coords = np.array([ax.get_position().get_points().ravel() for ax in axes])
     minx = np.min(coords[:,0])
     maxx = np.max(coords[:,2])
@@ -87,8 +94,15 @@ def common_ylabels(fig, *axes):
         axes = fig.get_axes()
     if len(axes) == 0:
         return
-    if len(axes) == 1 and isinstance(axes[0], np.ndarray):
-        axes = axes[0].ravel()
+    axs = []
+    for ax in axes:
+        if isinstance(ax, np.ndarray):
+            axs.extend(ax.ravel())
+        elif isinstance(ax, (tuple, list)):
+            axs.extend(ax)
+        else:
+            axs.append(ax)
+    axes = axs
     coords = np.array([ax.get_position().get_points().ravel() for ax in axes])
     # center common ylabel:
     minx = np.min(coords[:,0])
@@ -129,8 +143,15 @@ def common_xticks(fig, *axes):
         axes = fig.get_axes()
     if len(axes) == 0:
         return
-    if len(axes) == 1 and isinstance(axes[0], np.ndarray):
-        axes = axes[0].ravel()
+    axs = []
+    for ax in axes:
+        if isinstance(ax, np.ndarray):
+            axs.extend(ax.ravel())
+        elif isinstance(ax, (tuple, list)):
+            axs.extend(ax)
+        else:
+            axs.append(ax)
+    axes = axs
     coords = np.array([ax.get_position().get_points().ravel() for ax in axes])
     minx = np.min(coords[:,0])
     maxx = np.max(coords[:,2])
@@ -173,8 +194,15 @@ def common_yticks(fig, *axes):
         axes = fig.get_axes()
     if len(axes) == 0:
         return
-    if len(axes) == 1 and isinstance(axes[0], np.ndarray):
-        axes = axes[0].ravel()
+    axs = []
+    for ax in axes:
+        if isinstance(ax, np.ndarray):
+            axs.extend(ax.ravel())
+        elif isinstance(ax, (tuple, list)):
+            axs.extend(ax)
+        else:
+            axs.append(ax)
+    axes = axs
     coords = np.array([ax.get_position().get_points().ravel() for ax in axes])
     minx = np.min(coords[:,0])
     maxx = np.max(coords[:,2])
@@ -215,8 +243,15 @@ def common_xspines(fig, *axes):
         axes = fig.get_axes()
     if len(axes) == 0:
         return
-    if len(axes) == 1 and isinstance(axes[0], np.ndarray):
-        axes = axes[0].ravel()
+    axs = []
+    for ax in axes:
+        if isinstance(ax, np.ndarray):
+            axs.extend(ax.ravel())
+        elif isinstance(ax, (tuple, list)):
+            axs.extend(ax)
+        else:
+            axs.append(ax)
+    axes = axs
     coords = np.array([ax.get_position().get_points().ravel() for ax in axes])
     minx = np.min(coords[:,0])
     maxx = np.max(coords[:,2])
@@ -260,8 +295,15 @@ def common_yspines(fig, *axes):
         axes = fig.get_axes()
     if len(axes) == 0:
         return
-    if len(axes) == 1 and isinstance(axes[0], np.ndarray):
-        axes = axes[0].ravel()
+    axs = []
+    for ax in axes:
+        if isinstance(ax, np.ndarray):
+            axs.extend(ax.ravel())
+        elif isinstance(ax, (tuple, list)):
+            axs.extend(ax)
+        else:
+            axs.append(ax)
+    axes = axs
     coords = np.array([ax.get_position().get_points().ravel() for ax in axes])
     minx = np.min(coords[:,0])
     maxx = np.max(coords[:,2])