make script work for individual (single) groups

Author: knarrff

group_composition_plot/group_composition_plot.py

@@ -44,18 +44,18 @@ def errorexit(msg='internal error', code=1):
 # Creating the joint plot
 scaling = .9
 iymax = (len(data)+1)//2
-fig, axs = plt.subplots(iymax, 2, figsize=(scaling*16, scaling*8*iymax))
+fig = plt.figure(figsize=(scaling*16, scaling*8*iymax))
 
-ix = iy = 0
+i = 1
 for inst, idata in data.items():
-    axs[iy, ix].pie(idata['activity_weights'], colors=colors, startangle=140)
+    ax = fig.add_subplot(iymax, min(2, len(data)), i)
+    ax.pie(idata['activity_weights'], colors=colors, startangle=140)
     inst_name = idata['institution_name']
     if 'group_name' in idata:
         inst_name = idata['group_name'] + "\n" + inst_name
     if not args['hide_titles']:
-        axs[iy, ix].set_title(inst_name, fontsize=20)
-    iy += ix
-    ix = 1 if ix == 0 else 0
+        ax.set_title(inst_name, fontsize=20)
+    i += 1
 
 # Shared legend
 if args['legend']: