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", )