Change group figure generation

arfc · Feb 11, 2025 · b95c9a0 · b95c9a0
1 parent 8e93be5
commit b95c9a0
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/docs/figures/group_fig.py b/docs/figures/group_fig.py
@@ -47,5 +47,23 @@
 ax.legend(fontsize=12)
 plt.tight_layout()
 
+#plt.savefig('group_bar.png')
+plt.close()
+
+markers = ['v', '^', 'v', '^']
+colors = ['b', 'b', 'orange', 'orange']
+count = 0
+for key, val in data.items():
+    x = [1, 2, 3, 4, 5, 6]
+    size = [20] * 6
+    plt.scatter(x, val, marker=markers[count], c=colors[count],
+                label=key, sizes=size)
+    count += 1
+
+plt.ylabel('Absolute Percent Difference [%]')
+plt.xlabel('Group')
+plt.legend(fontsize=12)
+plt.tight_layout()
+
 plt.savefig('group_bar.png')
 plt.close()