def rotate(x, y, U_direction_radians): x_r = x*np.cos(U_direction_radians)-y*np.sin(U_direction_radians) y_r = x*np.sin(U_direction_radians)+y*np.cos(U_direction_radians) return x_r, y_r