Add parameters to plotting scripts

Add `imgname` parameter to be able to save a plot to a file instead of
showing it. Use-case: platforms without GUI like docker containers.

Add `labels` parameter to histogram plot. Use-case: replacing long
commands with human friendly labels.
This commit is contained in:
Benjamin Rosemann authored and David Peter committed 2021-02-16 08:21:45 +01:00
1 parent 1f7284dae3
commit b04a9ec6c4
3 files changed
+23 -5

No files matched your search

+5 -1
View File
@@ -18,6 +18,7 @@ parser.add_argument("--title", help="Plot Title")
parser.add_argument(
"--labels", help="Comma-separated list of entries for the plot legend"
)
parser.add_argument("--imgname", help="Save image to the given filename.")
args = parser.parse_args()
with open(args.file) as f:
@@ -41,4 +42,7 @@ if args.title:
plt.legend(handles=boxplot["boxes"], labels=labels, loc="best", fontsize="medium")
plt.ylabel("Time [s]")
plt.ylim(0, None)
plt.show()
if args.imgname:
plt.savefig(args.imgname)
else:
plt.show()