diff --git a/liveview.py b/liveview.py index 3aa9209..fcaf5eb 100644 --- a/liveview.py +++ b/liveview.py @@ -12,7 +12,7 @@ class ResistanceView: self.fig.canvas.flush_events() class VectorView: - def __init__(self, vectormap): + def __init__(self, vectormap, title=""): # Vector origin location X = np.repeat(range(9), 10) Y = np.resize(range(10), 10*9) @@ -23,10 +23,10 @@ class VectorView: # Creating plot pyplot.quiver(X, Y, U, V, color='b', units='xy', scale=1) - pyplot.title('Single Vector') + pyplot.title(title) ## x-lim and y-lim - pyplot.xlim(-1, 10) + pyplot.xlim(-1, 9) pyplot.ylim(-2, 11) # Show plot with grid