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])
