diff --git a/triangle/plot.py b/triangle/plot.py index 8155e2e..b312392 100644 --- a/triangle/plot.py +++ b/triangle/plot.py @@ -1,12 +1,14 @@ import numpy as np -def compare(plt, A, B, figsize=(6, 3)): +def compare(plt, A, B, figsize=(6, 3), title1="Input Points", title2="Triangulation"): plt.figure(figsize=figsize) ax1 = plt.subplot(121) + ax1.set_title(title1) plot(ax1, **A) lim = ax1.axis() ax2 = plt.subplot(122, sharey=ax1) + ax2.set_title(title2) plot(ax2, **B) ax2.axis(lim) plt.tight_layout()