The show()
method of the Particelset
class is capable of plotting the particle locations and velocity fields in scalar and vector form. In this notebook, we demonstrate these capabilities using the GlobCurrent dataset. We begin by importing the relevant modules.
from parcels import FieldSet, ParticleSet, JITParticle, AdvectionRK4
from datetime import timedelta, datetime
We then instatiate a FieldSet
with the velocity field data from GlobCurrent dataset.
filenames = {'U': "GlobCurrent_example_data/20*.nc",
'V': "GlobCurrent_example_data/20*.nc"}
variables = {'U': 'eastward_eulerian_current_velocity',
'V': 'northward_eulerian_current_velocity'}
dimensions = {'lat': 'lat',
'lon': 'lon',
'time': 'time'}
fieldset = FieldSet.from_netcdf(filenames, variables, dimensions)
WARNING: Casting lon data to np.float32 WARNING: Casting lat data to np.float32 WARNING: Casting lon data to np.float32 WARNING: Casting lat data to np.float32
Next, we instantiate a ParticeSet
composed of JITParticles
:
pset = ParticleSet.from_line(fieldset=fieldset, size=5, pclass=JITParticle,
start=(31, -31), finish=(32, -31), time=datetime(2002, 1, 1))
Given this ParticleSet
, we can now explore the different features of the show()
method. To start, let's simply call show()
with no arguments.
pset.show()
Then, let's advect the particles starting on January 1, 2002 for a week.
pset.execute(AdvectionRK4, runtime=timedelta(days=7),
dt=timedelta(minutes=5), moviedt=timedelta(hours=6))
Compiled JITParticleAdvectionRK4 ==> /var/folders/rn/w27yh9t521vdz7zq_lngqq1m0000gp/T/parcels-502/c40f5f0ead2b93182413ae905e4b193d.so
If we call show()
again, we will see that the particles have been advected:
pset.show()
If we want to save the file rather than show it, we set the argument savefile
equal to the 'path/to/save/file'
.
pset.show(savefile='particles')
Plot saved to particles.png
To set the domain of the plot, we specify the domain argument. The format domain
expects is [max lat, min lat, max lon, min lon]
. Note that the plotted domain is found by interpolating the user-specified domain onto the velocity grid. For instance,
pset.show(domain=[-30, -35, 35, 26])
We can also easily display a scalar contour plot of a single component of the velocity vector field. This is done by setting the field
argument equal to the desired scalar velocity field.
pset.show(field=fieldset.U)
To plot the scalar U velocity field at a different date and time, we set the argument t
equal to a datetime
or timedelta
object or simply the number of seconds since the time origin. For instance, let's view the U field on January, 10, 2002 at 2 PM.
pset.show(field=fieldset.U, show_time=datetime(2002, 1, 10, 2))
Note that the particle locations do not change, but remain at the locations corresponding to the end of the last integration. To remove them from the plot, we set the argument particles
equal to False
.
pset.show(field=fieldset.U, show_time=datetime(2002, 1, 10, 2), particles=False)
By setting the field
argument equal to vector
, we can display the velocity in full vector form
pset.show(field='vector')
The normalized vector field is colored by speed. To control the maximum speed value on the colorbar, set the vmax
argument equal to the desired value.
pset.show(field='vector', vmax=3.0, domain=[-30, -36, 33, 10])
We can also easily show the land present in the domain using the Basemap module's built-in land mask:
pset.show(field='vector', land=True, particles=False)