diff --git a/plot_stacked_bandwidth.py b/plot_stacked_bandwidth.py index 24dff6b..16040eb 100755 --- a/plot_stacked_bandwidth.py +++ b/plot_stacked_bandwidth.py @@ -55,4 +55,7 @@ if __name__ == "__main__": L.get_texts()[4].set_text("4G SCC 3") L.get_texts()[5].set_text("4G SCC 4") - plt.show() + if args.save: + plt.savefig("{}-used_bandwidth".format(args.save)) + else: + plt.show()