From 4cedd644cc3ea600d7bc5bc1e86d6854af400d7e Mon Sep 17 00:00:00 2001 From: Kasra Shirvanian Date: Sun, 16 Feb 2025 23:50:20 +0000 Subject: [PATCH] replace xarray median and mean with numpy --- examples/compute_polar_coordinates.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/examples/compute_polar_coordinates.py b/examples/compute_polar_coordinates.py index dd849c7c1..cf9271e73 100644 --- a/examples/compute_polar_coordinates.py +++ b/examples/compute_polar_coordinates.py @@ -251,16 +251,18 @@ rho_data.plot.hist(bins=50, ax=ax, edgecolor="lightgray", linewidth=0.5) # add mean +rho_mean = np.nanmean(rho_data) ax.axvline( - x=rho_data.mean().values, + x=rho_mean, c="b", linestyle="--", ) # add median +rho_median = np.nanmedian(rho_data) ax.axvline( - x=rho_data.median().values, + x=rho_median, c="r", linestyle="-", ) @@ -268,8 +270,8 @@ # add legend ax.legend( [ - f"mean = {np.nanmean(rho_data):.2f} pixels", - f"median = {np.nanmedian(rho_data):.2f} pixels", + f"mean = {rho_mean:.2f} pixels", + f"median = {rho_median:.2f} pixels", ], loc="best", )