def plot(self,
projection='global',
resolution='l',
continent_fill_color='0.9',
water_fill_color='1.0',
label='magnitude',
color='depth',
colormap=None,
show=True,
outfile=None,
method=None,
fig=None,
title=None,
**kwargs): # @UnusedVariable
"""
Creates preview map of all events in current Catalog object.
:type projection: str, optional
:param projection: The map projection. Currently supported are:
* ``"global"`` (Will plot the whole world.)
* ``"ortho"`` (Will center around the mean lat/long.)
* ``"local"`` (Will plot around local events)
Defaults to "global"
:type resolution: str, optional
:param resolution: Resolution of the boundary database to use. Will be
based directly to the basemap module. Possible values are:
* ``"c"`` (crude)
* ``"l"`` (low)
* ``"i"`` (intermediate)
* ``"h"`` (high)
* ``"f"`` (full)
Defaults to ``"l"``
:type continent_fill_color: Valid matplotlib color, optional
:param continent_fill_color: Color of the continents. Defaults to
``"0.9"`` which is a light gray.
:type water_fill_color: Valid matplotlib color, optional
:param water_fill_color: Color of all water bodies.
Defaults to ``"white"``.
:type label: str, optional
:param label: Events will be labelled based on the chosen property.
Possible values are:
* ``"magnitude"``
* ``None``
Defaults to ``"magnitude"``
:type color: str, optional
:param color: The events will be color-coded based on the chosen
property. Possible values are:
* ``"date"``
* ``"depth"``
Defaults to ``"depth"``
:type colormap: str, any matplotlib colormap, optional
:param colormap: The colormap for color-coding the events.
The event with the smallest property will have the
color of one end of the colormap and the event with the biggest
property the color of the other end with all other events in
between.
Defaults to None which will use the default colormap for the date
encoding and a colormap going from green over yellow to red for the
depth encoding.
:type show: bool
:param show: Whether to show the figure after plotting or not. Can be
used to do further customization of the plot before
showing it. Has no effect if `outfile` is specified.
:type outfile: str
:param outfile: Output file path to directly save the resulting image
(e.g. ``"/tmp/image.png"``). Overrides the ``show`` option, image
will not be displayed interactively. The given path/filename is
also used to automatically determine the output format. Supported
file formats depend on your matplotlib backend. Most backends
support png, pdf, ps, eps and svg. Defaults to ``None``.
The figure is closed after saving it to file.
:type method: str
:param method: Method to use for plotting. Possible values are:
* ``'basemap'`` to use the Basemap library
* ``'cartopy'`` to use the Cartopy library
* ``None`` to pick the best available library
Defaults to ``None``.
:type fig: :class:`matplotlib.figure.Figure` (or
:class:`matplotlib.axes.Axes`)
:param fig: Figure instance to reuse, returned from a previous
inventory/catalog plot call with `method=basemap`.
If a previous basemap plot is reused, any kwargs regarding the
basemap plot setup will be ignored (i.e. `projection`,
`resolution`, `continent_fill_color`, `water_fill_color`). Note
that multiple plots using colorbars likely are problematic, but
e.g. one station plot (without colorbar) and one event plot (with
colorbar) together should work well.
If an :class:`~matplotlib.axes.Axes` is supplied, the given axes is
used to plot into and no colorbar will be produced.
:type title: str
:param title: Title above plot. If left ``None``, an automatic title
will be generated. Set to ``""`` for no title.
:returns: Figure instance with the plot.
.. rubric:: Examples
Mollweide projection for global overview:
>>> from obspy import read_events
>>> cat = read_events()
>>> cat.plot() # doctest:+SKIP
.. plot::
from obspy import read_events
cat = read_events()
cat.plot()
Orthographic projection:
>>> cat.plot(projection="ortho") # doctest:+SKIP
.. plot::
from obspy import read_events
cat = read_events()
cat.plot(projection="ortho")
Local (Albers equal area) projection:
>>> cat.plot(projection="local") # doctest:+SKIP
.. plot::
from obspy import read_events
cat = read_events()
cat.plot(projection="local")
Combining a station and event plot (uses basemap):
>>> from obspy import read_inventory, read_events
>>> inv = read_inventory()
>>> cat = read_events()
>>> fig = inv.plot(method=basemap, show=False) # doctest:+SKIP
>>> cat.plot(method=basemap, fig=fig) # doctest:+SKIP
.. plot::
from obspy import read_inventory, read_events
inv = read_inventory()
cat = read_events()
fig = inv.plot(show=False)
cat.plot(fig=fig)
"""
from obspy.imaging.maps import plot_map, _plot_basemap_into_axes
import matplotlib
import matplotlib.pyplot as plt
if color not in ('date', 'depth'):
raise ValueError('Events can be color coded by date or depth. '
"'%s' is not supported." % (color, ))
if label not in (None, 'magnitude', 'depth'):
raise ValueError('Events can be labeled by magnitude or events can'
' not be labeled. '
"'%s' is not supported." % (label, ))
# lat/lon coordinates, magnitudes, dates
lats = []
lons = []
labels = []
mags = []
colors = []
times = []
for event in self:
if not event.origins:
msg = ("Event '%s' does not have an origin and will not be "
"plotted." % str(event.resource_id))
warnings.warn(msg)
continue
if not event.magnitudes:
msg = ("Event '%s' does not have a magnitude and will not be "
"plotted." % str(event.resource_id))
warnings.warn(msg)
continue
origin = event.preferred_origin() or event.origins[0]
lats.append(origin.latitude)
lons.append(origin.longitude)
times.append(origin.time)
magnitude = event.preferred_magnitude() or event.magnitudes[0]
mag = magnitude.mag
mags.append(mag)
labels.append((' %.1f' %
mag) if mag and label == 'magnitude' else '')
if color == 'date':
c_ = origin.get('time') or np.nan
else:
c_ = (origin.get('depth') or np.nan) / 1e3
colors.append(c_)
# Create the colormap for date based plotting.
if colormap is None:
colormap = obspy_sequential
if title is None:
if len(lons) > 1:
# if we have a `None` in the origin time list it likely ends up
# as min and/or max and causes problems..
times_ = np.ma.masked_equal(times, None).compressed()
min_time = times_.min()
max_time = times_.max()
title = (
"{event_count} events ({start} to {end}) "
"- Color codes {colorcode}, size the magnitude".format(
event_count=len(self.events),
start=min_time.strftime("%Y-%m-%d"),
end=max_time.strftime("%Y-%m-%d"),
colorcode="origin time"
if color == "date" else "depth"))
else:
title = "Event at %s" % times[0].strftime("%Y-%m-%d")
if color not in ("date", "depth"):
msg = "Invalid option for 'color' parameter (%s)." % color
raise ValueError(msg)
min_size = 2
max_size = 30
min_size_ = min(mags) - 1
max_size_ = max(mags) + 1
if len(lons) > 1:
frac = [(0.2 + (_i - min_size_)) / (max_size_ - min_size_)
for _i in mags]
size_plot = [(_i * (max_size - min_size))**2 for _i in frac]
else:
size_plot = 15.0**2
if isinstance(fig, matplotlib.axes.Axes):
if method is not None and method != "basemap":
msg = ("Plotting into an matplotlib.axes.Axes instance "
"currently only implemented for `method='basemap'`.")
raise NotImplementedError(msg)
ax = fig
fig = ax.figure
_plot_basemap_into_axes(ax=ax,
lons=lons,
lats=lats,
size=size_plot,
color=colors,
bmap=None,
labels=labels,
projection=projection,
resolution=resolution,
continent_fill_color=continent_fill_color,
water_fill_color=water_fill_color,
colormap=colormap,
marker="o",
title=title,
show=False,
**kwargs)
else:
fig = plot_map(method,
lons,
lats,
size_plot,
colors,
labels,
projection=projection,
resolution=resolution,
continent_fill_color=continent_fill_color,
water_fill_color=water_fill_color,
colormap=colormap,
marker="o",
title=title,
show=False,
fig=fig,
**kwargs)
if outfile:
fig.savefig(outfile)
plt.close(fig)
else:
if show:
plt.show()
return fig
def plot(self, projection='global', resolution='l',
continent_fill_color='0.9', water_fill_color='1.0',
label='magnitude', color='depth', colormap=None, show=True,
outfile=None, method=None, fig=None, title=None,
**kwargs): # @UnusedVariable
"""
Creates preview map of all events in current Catalog object.
:type projection: str, optional
:param projection: The map projection. Currently supported are:
* ``"global"`` (Will plot the whole world.)
* ``"ortho"`` (Will center around the mean lat/long.)
* ``"local"`` (Will plot around local events)
Defaults to "global"
:type resolution: str, optional
:param resolution: Resolution of the boundary database to use. Will be
based directly to the basemap module. Possible values are:
* ``"c"`` (crude)
* ``"l"`` (low)
* ``"i"`` (intermediate)
* ``"h"`` (high)
* ``"f"`` (full)
Defaults to ``"l"``
:type continent_fill_color: Valid matplotlib color, optional
:param continent_fill_color: Color of the continents. Defaults to
``"0.9"`` which is a light gray.
:type water_fill_color: Valid matplotlib color, optional
:param water_fill_color: Color of all water bodies.
Defaults to ``"white"``.
:type label: str, optional
:param label: Events will be labelled based on the chosen property.
Possible values are:
* ``"magnitude"``
* ``None``
Defaults to ``"magnitude"``
:type color: str, optional
:param color: The events will be color-coded based on the chosen
property. Possible values are:
* ``"date"``
* ``"depth"``
Defaults to ``"depth"``
:type colormap: str, any matplotlib colormap, optional
:param colormap: The colormap for color-coding the events.
The event with the smallest property will have the
color of one end of the colormap and the event with the biggest
property the color of the other end with all other events in
between.
Defaults to None which will use the default colormap for the date
encoding and a colormap going from green over yellow to red for the
depth encoding.
:type show: bool
:param show: Whether to show the figure after plotting or not. Can be
used to do further customization of the plot before
showing it. Has no effect if `outfile` is specified.
:type outfile: str
:param outfile: Output file path to directly save the resulting image
(e.g. ``"/tmp/image.png"``). Overrides the ``show`` option, image
will not be displayed interactively. The given path/filename is
also used to automatically determine the output format. Supported
file formats depend on your matplotlib backend. Most backends
support png, pdf, ps, eps and svg. Defaults to ``None``.
The figure is closed after saving it to file.
:type method: str
:param method: Method to use for plotting. Possible values are:
* ``'basemap'`` to use the Basemap library
* ``'cartopy'`` to use the Cartopy library
* ``None`` to pick the best available library
Defaults to ``None``.
:type fig: :class:`matplotlib.figure.Figure` (or
:class:`matplotlib.axes.Axes`)
:param fig: Figure instance to reuse, returned from a previous
inventory/catalog plot call with `method=basemap`.
If a previous basemap plot is reused, any kwargs regarding the
basemap plot setup will be ignored (i.e. `projection`,
`resolution`, `continent_fill_color`, `water_fill_color`). Note
that multiple plots using colorbars likely are problematic, but
e.g. one station plot (without colorbar) and one event plot (with
colorbar) together should work well.
If an :class:`~matplotlib.axes.Axes` is supplied, the given axes is
used to plot into and no colorbar will be produced.
:type title: str
:param title: Title above plot. If left ``None``, an automatic title
will be generated. Set to ``""`` for no title.
:returns: Figure instance with the plot.
.. rubric:: Examples
Mollweide projection for global overview:
>>> from obspy import read_events
>>> cat = read_events()
>>> cat.plot() # doctest:+SKIP
.. plot::
from obspy import read_events
cat = read_events()
cat.plot()
Orthographic projection:
>>> cat.plot(projection="ortho") # doctest:+SKIP
.. plot::
from obspy import read_events
cat = read_events()
cat.plot(projection="ortho")
Local (Albers equal area) projection:
>>> cat.plot(projection="local") # doctest:+SKIP
.. plot::
from obspy import read_events
cat = read_events()
cat.plot(projection="local")
Combining a station and event plot (uses basemap):
>>> from obspy import read_inventory, read_events
>>> inv = read_inventory()
>>> cat = read_events()
>>> fig = inv.plot(method=basemap, show=False) # doctest:+SKIP
>>> cat.plot(method=basemap, fig=fig) # doctest:+SKIP
.. plot::
from obspy import read_inventory, read_events
inv = read_inventory()
cat = read_events()
fig = inv.plot(show=False)
cat.plot(fig=fig)
"""
from obspy.imaging.maps import plot_map, _plot_basemap_into_axes
import matplotlib
import matplotlib.pyplot as plt
if color not in ('date', 'depth'):
raise ValueError('Events can be color coded by date or depth. '
"'%s' is not supported." % (color,))
if label not in (None, 'magnitude', 'depth'):
raise ValueError('Events can be labeled by magnitude or events can'
' not be labeled. '
"'%s' is not supported." % (label,))
# lat/lon coordinates, magnitudes, dates
lats = []
lons = []
labels = []
mags = []
colors = []
times = []
for event in self:
if not event.origins:
msg = ("Event '%s' does not have an origin and will not be "
"plotted." % str(event.resource_id))
warnings.warn(msg)
continue
if not event.magnitudes:
msg = ("Event '%s' does not have a magnitude and will not be "
"plotted." % str(event.resource_id))
warnings.warn(msg)
continue
origin = event.preferred_origin() or event.origins[0]
lats.append(origin.latitude)
lons.append(origin.longitude)
times.append(origin.time)
magnitude = event.preferred_magnitude() or event.magnitudes[0]
mag = magnitude.mag
mags.append(mag)
labels.append((' %.1f' % mag) if mag and label == 'magnitude'
else '')
if color == 'date':
c_ = origin.get('time') or np.nan
else:
c_ = (origin.get('depth') or np.nan) / 1e3
colors.append(c_)
# Create the colormap for date based plotting.
if colormap is None:
colormap = obspy_sequential
if title is None:
if len(lons) > 1:
# if we have a `None` in the origin time list it likely ends up
# as min and/or max and causes problems..
times_ = np.ma.masked_equal(times, None).compressed()
min_time = times_.min()
max_time = times_.max()
title = (
"{event_count} events ({start} to {end}) "
"- Color codes {colorcode}, size the magnitude".format(
event_count=len(self.events),
start=min_time.strftime("%Y-%m-%d"),
end=max_time.strftime("%Y-%m-%d"),
colorcode="origin time"
if color == "date"
else "depth"))
else:
title = "Event at %s" % times[0].strftime("%Y-%m-%d")
if color not in ("date", "depth"):
msg = "Invalid option for 'color' parameter (%s)." % color
raise ValueError(msg)
min_size = 2
max_size = 30
min_size_ = min(mags) - 1
max_size_ = max(mags) + 1
if len(lons) > 1:
frac = [(0.2 + (_i - min_size_)) / (max_size_ - min_size_)
for _i in mags]
size_plot = [(_i * (max_size - min_size)) ** 2 for _i in frac]
else:
size_plot = 15.0 ** 2
if isinstance(fig, matplotlib.axes.Axes):
if method is not None and method != "basemap":
msg = ("Plotting into an matplotlib.axes.Axes instance "
"currently only implemented for `method='basemap'`.")
raise NotImplementedError(msg)
ax = fig
fig = ax.figure
_plot_basemap_into_axes(
ax=ax, lons=lons, lats=lats, size=size_plot,
color=colors, bmap=None, labels=labels,
projection=projection, resolution=resolution,
continent_fill_color=continent_fill_color,
water_fill_color=water_fill_color,
colormap=colormap, marker="o", title=title,
show=False, **kwargs)
else:
fig = plot_map(method, lons, lats, size_plot, colors, labels,
projection=projection, resolution=resolution,
continent_fill_color=continent_fill_color,
water_fill_color=water_fill_color,
colormap=colormap, marker="o", title=title,
show=False, fig=fig, **kwargs)
if outfile:
fig.savefig(outfile)
plt.close(fig)
else:
if show:
plt.show()
return fig