From d61e70b224a317f8709255a30926b33f4ca66fd9 Mon Sep 17 00:00:00 2001
From: Katelyn FitzGerald <7872563+kafitzgerald@users.noreply.github.com>
Date: Thu, 1 Feb 2024 13:41:40 -0700
Subject: [PATCH] add rotating and rescaling to the polar quiver example

---
 Gallery/Contours/NCL_polar_8.py | 24 ++++++++++++++----------
 1 file changed, 14 insertions(+), 10 deletions(-)

diff --git a/Gallery/Contours/NCL_polar_8.py b/Gallery/Contours/NCL_polar_8.py
index aa9c138dd..53bfdc132 100755
--- a/Gallery/Contours/NCL_polar_8.py
+++ b/Gallery/Contours/NCL_polar_8.py
@@ -25,21 +25,25 @@
 ###############################################################################
 # Read in data:
 
-# Open a netCDF data file using xarray default engine and load the data into xarrays
+# Open a netCDF data file using xarray default engine
 ds = xr.open_dataset(gdf.get("netcdf_files/atmos.nc"), decode_times=False)
-u = ds.U[0, 1, :, :]
-v = ds.V[0, 1, :, :]
-t = ds.TS[0, :, :]
+U = ds.U[0, 1, :, :].sel(lat=slice(60, 90))
+V = ds.V[0, 1, :, :].sel(lat=slice(60, 90))
+T = ds.TS[0, :, :].sel(lat=slice(60, 90))
 
-# Extract slices of data
-U = u.sel(lat=slice(60, 90))
-V = u.sel(lat=slice(60, 90))
-T = t.sel(lat=slice(60, 90))
+# Rotate and rescale the wind vectors following recommendations from SciTools/cartopy#1179
+U_source_crs = U / np.cos(U["lat"] / 180. * np.pi)
+V_source_crs = V
+magnitude = np.sqrt(U**2 + V**2)
+magnitude_source_crs = np.sqrt(U_source_crs**2 + V_source_crs**2)
+
+U_rs = U_source_crs * magnitude / magnitude_source_crs
+V_rs = V_source_crs * magnitude / magnitude_source_crs
 
 ###############################################################################
 # Fix the artifact of not-shown-data around 0 and 360-degree longitudes
-wrap_U = gv.xr_add_cyclic_longitudes(U, "lon")
-wrap_V = gv.xr_add_cyclic_longitudes(V, "lon")
+wrap_U = gv.xr_add_cyclic_longitudes(U_rs, "lon")
+wrap_V = gv.xr_add_cyclic_longitudes(V_rs, "lon")
 wrap_T = gv.xr_add_cyclic_longitudes(T, "lon")
 
 ###############################################################################