this repo has no description
at develop 1.1 kB view raw
1#!/usr/bin/env python3 2import sys 3 4import matplotlib as mpl 5import matplotlib.pyplot as plt 6import numpy as np 7import pandas as pd 8import seaborn as sns 9 10if __name__ == "__main__": 11 data = pd.read_csv(sys.argv[1]) 12 13 sns.set(font_scale=1.50, style="whitegrid", font="IBM Plex Sans") 14 fig, ax = plt.subplots() 15 16 plot = sns.scatterplot( 17 data=data, 18 x="Compile Time (s)", 19 y="Solve Time (s)", 20 hue="Configuration", 21 style="Configuration", 22 legend=True, 23 s=150, 24 ) 25 26 if "radiation" in sys.argv[1]: 27 ax.set_xlim(0, 0.5) 28 ax.set_ylim(0, 0.5) 29 # ax.legend(handles=handles, labels=labels) 30 # plot.legend(bbox_to_anchor=(0.23, 0.95), loc="upper left", borderaxespad=0) 31 elif "gbac" in sys.argv[1]: 32 handles, labels = ax.get_legend_handles_labels() 33 ax.legend(handles=handles, labels=labels) 34 plot.legend(bbox_to_anchor=(0.23, 0.95), loc="upper left", borderaxespad=0) 35 ax.set_xlim(0, 30) 36 ax.set_ylim(0, 30) 37 38 plot.figure.savefig("output.pdf", bbox_inches="tight")