def plot(ax: plt.axis, R, data, title=""): """ Plot function for subtask 1 """ ax.set_aspect('equal') arrowed_spines(ax) ax.set_title(title)
def plot(ax: plt.axis, x, y, u, v, trajectory, pause=0.2, title=""): """ Plot function for subtask 1 """ ax.set_aspect('equal') arrowed_spines(ax) ax.set_xlim([min(x) * 1, max(x) * 1]) ax.set_ylim([min(y) * 1, max(y) * 1]) ax.set_title(title) ax.streamplot(x, y, u, v) ax.plot(trajectory[:, 0], trajectory[:, 1])