diff --git a/Inst_eddies/Analysis/plot_velocity.py b/Inst_eddies/Analysis/plot_velocity.py index 4a3dd3fe48cc020553820762e84e98838e4f24ca..9092e9c63b3df1e5f643b8ba8a7617cac7157531 100755 --- a/Inst_eddies/Analysis/plot_velocity.py +++ b/Inst_eddies/Analysis/plot_velocity.py @@ -6,6 +6,7 @@ import sys import matplotlib.pyplot as plt import argparse import netCDF4 +import wind_cartopy parser = argparse.ArgumentParser() parser.add_argument("-s", "--scale", default = 20, type = float, @@ -70,11 +71,12 @@ with netCDF4.Dataset(args.input_file) as f: lat_2d[undef_velocity].reshape(-1), transform = src_crs, marker = "*", color = "violet", linestyle = "None") else: - quiver_return = ax.quiver(longitude[lon_mask], latitude[lat_mask], - ugos[lat_mask][:, lon_mask], - vgos[lat_mask][:, lon_mask], - scale = args.scale, scale_units = "width", - transform = src_crs) + quiver_return = wind_cartopy.plot(ax, longitude[lon_mask], + latitude[lat_mask], + ugos[lat_mask][:, lon_mask], + vgos[lat_mask][:, lon_mask], + scale = args.scale, + scale_units = "width") ax.quiverkey(quiver_return, 0.9, 0.9, 1, r"1 m s$^{-1}$", coordinates = "figure")