diff options
| -rwxr-xr-x | scripts/serd_bench.py | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/scripts/serd_bench.py b/scripts/serd_bench.py index f3c3e148..54ec3f29 100755 --- a/scripts/serd_bench.py +++ b/scripts/serd_bench.py @@ -115,18 +115,18 @@ def plot(in_file, out_filename, x_label, y_label, y_max=None): ax.set_xlabel(x_label) ax.set_ylabel(y_label) - if y_max is not None: - ax.set_ylim([0.0, y_max]) - ax.grid(linewidth=0.25, linestyle=":", color="0", dashes=[0.2, 1.6]) ax.ticklabel_format(style="sci", scilimits=(4, 0), useMathText=True) ax.tick_params(axis="both", width=0.75) x = list(map(float, cols[0])) + actual_y_max = 0.0 for i, y in enumerate(cols[1::]): + y_floats = list(map(float, y)) + actual_y_max = max(actual_y_max, max(y_floats)) ax.plot( x, - list(map(float, y)), + y_floats, label=header[i + 1], marker=next(markers), dashes=next(dashes), @@ -134,6 +134,9 @@ def plot(in_file, out_filename, x_label, y_label, y_max=None): linewidth=1.0, ) + y_max = actual_y_max if y_max is None else y_max + ax.set_ylim([0.0, y_max]) + plt.legend(labelspacing=0.25) plt.savefig(out_filename, bbox_inches="tight", pad_inches=0.125) plt.close() |