WcsMap¶
-
class
gammapy.maps.
WcsMap
(geom, data, meta=None, unit='')[source]¶ Bases:
gammapy.maps.Map
Base class for WCS map classes.
Attributes Summary
Data array (
ndarray
)Map geometry (
Geom
)Whether map is mask with bool dtype
Map meta (
dict
)Map data times unit (
Quantity
)Map unit (
Unit
)Methods Summary
apply_edisp
(edisp)Apply energy dispersion to map.
coadd
(map_in[, weights])Add the contents of
map_in
to this map.copy
(**kwargs)Copy map instance and overwrite given attributes, except for geometry.
create
([map_type, npix, binsz, width, proj, …])Factory method to create an empty WCS map.
crop
(crop_width)Crop the spatial dimensions of the map.
cumsum
(axis_name)Compute cumulative sum along a given axis
downsample
(factor[, preserve_counts, axis_name])Downsample the spatial dimension by a given factor.
fill_by_coord
(coords[, weights])Fill pixels at
coords
with givenweights
.fill_by_idx
(idx[, weights])Fill pixels at
idx
with givenweights
.fill_by_pix
(pix[, weights])Fill pixels at
pix
with givenweights
.fill_events
(events)Fill event coordinates (
EventList
).from_geom
(geom[, meta, data, unit, dtype])Generate an empty map from a
Geom
instance.from_hdulist
(hdu_list[, hdu, hdu_bands, format])Make a WcsMap object from a FITS HDUList.
from_stack
(maps[, axis, axis_name])Create Map from list of images and a non-spatial axis.
get_by_coord
(coords[, fill_value])Return map values at the given map coordinates.
get_by_idx
(idx)Return map values at the given pixel indices.
get_by_pix
(pix[, fill_value])Return map values at the given pixel coordinates.
get_image_by_coord
(coords)Return spatial map at the given axis coordinates.
get_image_by_idx
(idx)Return spatial map at the given axis pixel indices.
get_image_by_pix
(pix)Return spatial map at the given axis pixel coordinates
get_spectrum
([region, func, weights])Extract spectrum in a given region.
integral
(axis_name, coords, **kwargs)Compute integral along a given axis
interp_by_coord
(coords[, method, fill_value])Interpolate map values at the given map coordinates.
interp_by_pix
(pix[, method, fill_value])Interpolate map values at the given pixel coordinates.
interp_to_geom
(geom[, preserve_counts, …])Interpolate map to input geometry.
Iterate over image planes of the map.
mask_nearest_position
(position)Given a sky coordinate return nearest valid position in the mask
normalize
([axis_name])Normalise data in place along a given axis.
pad
(pad_width[, axis_name, mode, cval, method])Pad the spatial dimensions of the map.
plot_grid
([figsize, ncols])Plot map as a grid of subplots for non-spatial axes
plot_interactive
([rc_params])Plot map with interactive widgets to explore the non spatial axes.
read
(filename[, hdu, hdu_bands, map_type, …])Read a map from a FITS file.
reduce
(axis_name[, func, keepdims, weights])Reduce map over a single non-spatial axis
reduce_over_axes
([func, keepdims, …])Reduce map over non-spatial axes
resample_axis
(axis[, weights, ufunc])Resample map to a new axis binning by grouping over smaller bins and apply ufunc to the bin contents.
set_by_coord
(coords, vals)Set pixels at
coords
with givenvals
.set_by_idx
(idx, vals)Set pixels at
idx
with givenvals
.set_by_pix
(pix, vals)Set pixels at
pix
with givenvals
.slice_by_idx
(slices)Slice sub map from map object.
sum_over_axes
([axes_names, keepdims, weights])To sum map values over all non-spatial axes.
to_cube
(axes)Append non-spatial axes to create a higher-dimensional Map.
to_hdu
([hdu, hdu_bands, sparse])Make a FITS HDU from this map.
to_hdulist
([hdu, hdu_bands, sparse, format])Convert to
HDUList
.to_unit
(unit)Convert map to different unit
upsample
(factor[, order, preserve_counts, axis])Upsample the spatial dimension by a given factor.
write
(filename[, overwrite])Write to a FITS file.
Attributes Documentation
-
is_mask
¶ Whether map is mask with bool dtype
-
tag
= 'map'¶
Methods Documentation
-
apply_edisp
(edisp)¶ Apply energy dispersion to map. Requires energy axis.
- Parameters
- edisp
gammapy.irf.EDispKernel
Energy dispersion matrix
- edisp
- Returns
- map
WcsNDMap
Map with energy dispersion applied.
- map
-
coadd
(map_in, weights=None)¶ Add the contents of
map_in
to this map.This method can be used to combine maps containing integral quantities (e.g. counts) or differential quantities if the maps have the same binning.
- Parameters
- map_in
Map
Input map.
- weights: `Map` or `~numpy.ndarray`
The weight factors while adding
- map_in
-
copy
(**kwargs)¶ Copy map instance and overwrite given attributes, except for geometry.
- Parameters
- **kwargsdict
Keyword arguments to overwrite in the map constructor.
- Returns
- copy
Map
Copied Map.
- copy
-
classmethod
create
(map_type='wcs', npix=None, binsz=0.1, width=None, proj='CAR', frame='icrs', refpix=None, axes=None, skydir=None, dtype='float32', meta=None, unit='')[source]¶ Factory method to create an empty WCS map.
- Parameters
- map_type{‘wcs’, ‘wcs-sparse’}
Map type. Selects the class that will be used to instantiate the map.
- npixint or tuple or list
Width of the map in pixels. A tuple will be interpreted as parameters for longitude and latitude axes. For maps with non-spatial dimensions, list input can be used to define a different map width in each image plane. This option supersedes width.
- widthfloat or tuple or list
Width of the map in degrees. A tuple will be interpreted as parameters for longitude and latitude axes. For maps with non-spatial dimensions, list input can be used to define a different map width in each image plane.
- binszfloat or tuple or list
Map pixel size in degrees. A tuple will be interpreted as parameters for longitude and latitude axes. For maps with non-spatial dimensions, list input can be used to define a different bin size in each image plane.
- skydirtuple or
SkyCoord
Sky position of map center. Can be either a SkyCoord object or a tuple of longitude and latitude in deg in the coordinate system of the map.
- frame{“icrs”, “galactic”}, optional
Coordinate system, either Galactic (“galactic”) or Equatorial (“icrs”).
- axeslist
List of non-spatial axes.
- projstring, optional
Any valid WCS projection type. Default is ‘CAR’ (cartesian).
- refpixtuple
Reference pixel of the projection. If None then this will be chosen to be center of the map.
- dtypestr, optional
Data type, default is float32
- meta
dict
Dictionary to store meta data.
- unitstr or
Unit
The unit of the map
- Returns
- map
WcsMap
A WCS map object.
- map
-
abstract
crop
(crop_width)¶ Crop the spatial dimensions of the map.
-
cumsum
(axis_name)¶ Compute cumulative sum along a given axis
- Parameters
- axis_namestr
Along which axis to integrate.
- Returns
- cumsum
Map
Map with cumulative sum
- cumsum
-
abstract
downsample
(factor, preserve_counts=True, axis_name=None)¶ Downsample the spatial dimension by a given factor.
- Parameters
- factorint
Downsampling factor.
- preserve_countsbool
Preserve the integral over each bin. This should be true if the map is an integral quantity (e.g. counts) and false if the map is a differential quantity (e.g. intensity).
- axis_namestr
Which axis to downsample. By default spatial axes are downsampled.
- Returns
- map
Map
Downsampled map.
- map
-
fill_by_coord
(coords, weights=None)¶ Fill pixels at
coords
with givenweights
.
-
abstract
fill_by_idx
(idx, weights=None)¶ Fill pixels at
idx
with givenweights
.- Parameters
- idxtuple
Tuple of pixel index arrays for each dimension of the map. Tuple should be ordered as (I_lon, I_lat, I_0, …, I_n) for WCS maps and (I_hpx, I_0, …, I_n) for HEALPix maps.
- weights
ndarray
Weights vector. Default is weight of one.
-
fill_by_pix
(pix, weights=None)¶ Fill pixels at
pix
with givenweights
.- Parameters
- pixtuple
Tuple of pixel index arrays for each dimension of the map. Tuple should be ordered as (I_lon, I_lat, I_0, …, I_n) for WCS maps and (I_hpx, I_0, …, I_n) for HEALPix maps. Pixel indices can be either float or integer type. Float indices will be rounded to the nearest integer.
- weights
ndarray
Weights vector. Default is weight of one.
-
static
from_geom
(geom, meta=None, data=None, unit='', dtype='float32')¶ Generate an empty map from a
Geom
instance.- Parameters
- geom
Geom
Map geometry.
- data
numpy.ndarray
data array
- meta
dict
Dictionary to store meta data.
- unitstr or
Unit
Data unit.
- geom
- Returns
- map_out
Map
Map object
- map_out
-
classmethod
from_hdulist
(hdu_list, hdu=None, hdu_bands=None, format='gadf')[source]¶ Make a WcsMap object from a FITS HDUList.
-
classmethod
from_stack
(maps, axis=None, axis_name=None)¶ Create Map from list of images and a non-spatial axis.
The image geometries must be aligned, except for the axis that is stacked.
- Parameters
- Returns
- map
Map
Map with additional non-spatial axis.
- map
-
get_by_coord
(coords, fill_value=nan)¶ Return map values at the given map coordinates.
- Parameters
- coordstuple or
MapCoord
Coordinate arrays for each dimension of the map. Tuple should be ordered as (lon, lat, x_0, …, x_n) where x_i are coordinates for non-spatial dimensions of the map.
- fill_valuefloat
Value which is returned if the position is outside of the projection footprint
- coordstuple or
- Returns
- vals
ndarray
Values of pixels in the map. np.nan used to flag coords outside of map.
- vals
-
abstract
get_by_idx
(idx)¶ Return map values at the given pixel indices.
- Parameters
- idxtuple
Tuple of pixel index arrays for each dimension of the map. Tuple should be ordered as (I_lon, I_lat, I_0, …, I_n) for WCS maps and (I_hpx, I_0, …, I_n) for HEALPix maps.
- Returns
- vals
ndarray
Array of pixel values. np.nan used to flag coordinate outside of map
- vals
-
get_by_pix
(pix, fill_value=nan)¶ Return map values at the given pixel coordinates.
- Parameters
- pixtuple
Tuple of pixel index arrays for each dimension of the map. Tuple should be ordered as (I_lon, I_lat, I_0, …, I_n) for WCS maps and (I_hpx, I_0, …, I_n) for HEALPix maps. Pixel indices can be either float or integer type.
- fill_valuefloat
Value which is returned if the position is outside of the projection footprint
- Returns
- vals
ndarray
Array of pixel values. np.nan used to flag coordinates outside of map
- vals
-
get_image_by_coord
(coords)¶ Return spatial map at the given axis coordinates.
- Parameters
- coordstuple or dict
Tuple should be ordered as (x_0, …, x_n) where x_i are coordinates for non-spatial dimensions of the map. Dict should specify the axis names of the non-spatial axes such as {‘axes0’: x_0, …, ‘axesn’: x_n}.
- Returns
- map_out
Map
Map with spatial dimensions only.
- map_out
See also
Examples
import numpy as np from gammapy.maps import Map, MapAxis from astropy.coordinates import SkyCoord from astropy import units as u # Define map axes energy_axis = MapAxis.from_edges( np.logspace(-1., 1., 4), unit='TeV', name='energy', ) time_axis = MapAxis.from_edges( np.linspace(0., 10, 20), unit='h', name='time', ) # Define map center skydir = SkyCoord(0, 0, frame='galactic', unit='deg') # Create map m_wcs = Map.create( map_type='wcs', binsz=0.02, skydir=skydir, width=10.0, axes=[energy_axis, time_axis], ) # Get image by coord tuple image = m_wcs.get_image_by_coord(('500 GeV', '1 h')) # Get image by coord dict with strings image = m_wcs.get_image_by_coord({'energy': '500 GeV', 'time': '1 h'}) # Get image by coord dict with quantities image = m_wcs.get_image_by_coord({'energy': 0.5 * u.TeV, 'time': 1 * u.h})
-
get_image_by_idx
(idx)¶ Return spatial map at the given axis pixel indices.
- Parameters
- idxtuple
Tuple of scalar indices for each non spatial dimension of the map. Tuple should be ordered as (I_0, …, I_n).
- Returns
- map_out
Map
Map with spatial dimensions only.
- map_out
See also
-
get_image_by_pix
(pix)¶ Return spatial map at the given axis pixel coordinates
- Parameters
- pixtuple
Tuple of scalar pixel coordinates for each non-spatial dimension of the map. Tuple should be ordered as (I_0, …, I_n). Pixel coordinates can be either float or integer type.
- Returns
- map_out
Map
Map with spatial dimensions only.
- map_out
See also
-
get_spectrum
(region=None, func=<function nansum>, weights=None)¶ Extract spectrum in a given region.
The spectrum can be computed by summing (or, more generally, applying
func
) along the spatial axes in each energy bin. This occurs only inside theregion
, which by default is assumed to be the whole spatial extension of the map.- Parameters
- region: `~regions.Region`
Region (pixel or sky regions accepted).
- funcnumpy.func
Function to reduce the data. Default is np.nansum. For a boolean Map, use np.any or np.all.
- weights
WcsNDMap
Array to be used as weights. The geometry must be equivalent.
- Returns
- spectrum
RegionNDMap
Spectrum in the given region.
- spectrum
-
integral
(axis_name, coords, **kwargs)¶ Compute integral along a given axis
This method uses interpolation of the cumulative sum.
-
abstract
interp_by_coord
(coords, method='nearest', fill_value=None)¶ Interpolate map values at the given map coordinates.
- Parameters
- coordstuple or
MapCoord
Coordinate arrays for each dimension of the map. Tuple should be ordered as (lon, lat, x_0, …, x_n) where x_i are coordinates for non-spatial dimensions of the map.
- method{“nearest”, “linear”}
Method to interpolate data values. By default no interpolation is performed and the return value will be the amplitude of the pixel encompassing the given coordinate.
- fill_valueNone or float value
The value to use for points outside of the interpolation domain. If None, values outside the domain are extrapolated.
- coordstuple or
- Returns
- vals
ndarray
Interpolated pixel values.
- vals
-
abstract
interp_by_pix
(pix, method='nearest', fill_value=None)¶ Interpolate map values at the given pixel coordinates.
- Parameters
- pixtuple
Tuple of pixel coordinate arrays for each dimension of the map. Tuple should be ordered as (p_lon, p_lat, p_0, …, p_n) where p_i are pixel coordinates for non-spatial dimensions of the map.
- method{“nearest”, “linear”}
Method to interpolate data values. By default no interpolation is performed and the return value will be the amplitude of the pixel encompassing the given coordinate.
- fill_valueNone or float value
The value to use for points outside of the interpolation domain. If None, values outside the domain are extrapolated.
- Returns
- vals
ndarray
Interpolated pixel values.
- vals
-
interp_to_geom
(geom, preserve_counts=False, fill_value=0, **kwargs)¶ Interpolate map to input geometry.
- Parameters
- geom
Geom
Target Map geometry
- preserve_countsbool
Preserve the integral over each bin. This should be true if the map is an integral quantity (e.g. counts) and false if the map is a differential quantity (e.g. intensity)
- **kwargsdict
Keyword arguments passed to
Map.interp_by_coord
- geom
- Returns
- interp_map
Map
Interpolated Map
- interp_map
-
iter_by_image
()¶ Iterate over image planes of the map.
This is a generator yielding
(data, idx)
tuples, wheredata
is anumpy.ndarray
view of the image plane data, andidx
is a tuple of int, the index of the image plane.The image plane index is in data order, so that the data array can be indexed directly.
-
mask_nearest_position
(position)¶ Given a sky coordinate return nearest valid position in the mask
If the mask contains additional axes, the mask is reduced over those.
- Parameters
- position
SkyCoord
Test position
- position
- Returns
- position
SkyCoord
Nearest position in the mask
- position
-
normalize
(axis_name=None)¶ Normalise data in place along a given axis.
- Parameters
- axis_namestr
Along which axis to normalize.
-
pad
(pad_width, axis_name=None, mode='constant', cval=0, method='linear')¶ Pad the spatial dimensions of the map.
- Parameters
- pad_width{sequence, array_like, int}
Number of pixels padded to the edges of each axis.
- axis_namestr
Which axis to downsample. By default spatial axes are padded.
- mode{‘edge’, ‘constant’, ‘interp’}
Padding mode. ‘edge’ pads with the closest edge value. ‘constant’ pads with a constant value. ‘interp’ pads with an extrapolated value.
- cvalfloat
Padding value when mode=’consant’.
- Returns
- map
Map
Padded map.
- map
-
plot_grid
(figsize=None, ncols=3, **kwargs)¶ Plot map as a grid of subplots for non-spatial axes
- Parameters
- figsizetuple of int
Figsize to plot on
- ncolsint
Number of columns to plot
- **kwargsdict
Keyword arguments passed to
Map.plot
.
- Returns
- axes
ndarray
ofAxes
Axes grid
- axes
-
plot_interactive
(rc_params=None, **kwargs)¶ Plot map with interactive widgets to explore the non spatial axes.
- Parameters
- rc_paramsdict
Passed to
matplotlib.rc_context(rc=rc_params)
to style the plot.- **kwargsdict
Keyword arguments passed to
WcsNDMap.plot
.
Examples
You can try this out e.g. using a Fermi-LAT diffuse model cube with an energy axis:
from gammapy.maps import Map m = Map.read("$GAMMAPY_DATA/fermi_3fhl/gll_iem_v06_cutout.fits") m.plot_interactive(add_cbar=True, stretch="sqrt")
If you would like to adjust the figure size you can use the
rc_params
argument:rc_params = {'figure.figsize': (12, 6), 'font.size': 12} m.plot_interactive(rc_params=rc_params)
-
static
read
(filename, hdu=None, hdu_bands=None, map_type='auto', format=None, colname=None)¶ Read a map from a FITS file.
- Parameters
- filenamestr or
Path
Name of the FITS file.
- hdustr
Name or index of the HDU with the map data.
- hdu_bandsstr
Name or index of the HDU with the BANDS table. If not defined this will be inferred from the FITS header of the map HDU.
- map_type{‘wcs’, ‘wcs-sparse’, ‘hpx’, ‘hpx-sparse’, ‘auto’, ‘region’}
Map type. Selects the class that will be used to instantiate the map. The map type should be consistent with the format of the input file. If map_type is ‘auto’ then an appropriate map type will be inferred from the input file.
- colnamestr, optional
data column name to be used of healix map.
- filenamestr or
- Returns
- map_out
Map
Map object
- map_out
-
reduce
(axis_name, func=<ufunc 'add'>, keepdims=False, weights=None)¶ Reduce map over a single non-spatial axis
- Parameters
- Returns
- map_out
Map
Map with the given non-spatial axes reduced
- map_out
-
reduce_over_axes
(func=<ufunc 'add'>, keepdims=False, axes_names=None, weights=None)¶ Reduce map over non-spatial axes
- Parameters
- Returns
- map_out
Map
Map with non-spatial axes reduced
- map_out
-
resample_axis
(axis, weights=None, ufunc=<ufunc 'add'>)¶ Resample map to a new axis binning by grouping over smaller bins and apply ufunc to the bin contents.
By default, the map content are summed over the smaller bins. Other numpy ufunc can be used, e.g. np.logical_and, np.logical_or
-
set_by_coord
(coords, vals)¶ Set pixels at
coords
with givenvals
.
-
abstract
set_by_idx
(idx, vals)¶ Set pixels at
idx
with givenvals
.- Parameters
- idxtuple
Tuple of pixel index arrays for each dimension of the map. Tuple should be ordered as (I_lon, I_lat, I_0, …, I_n) for WCS maps and (I_hpx, I_0, …, I_n) for HEALPix maps.
- vals
ndarray
Values vector.
-
set_by_pix
(pix, vals)¶ Set pixels at
pix
with givenvals
.- Parameters
- pixtuple
Tuple of pixel index arrays for each dimension of the map. Tuple should be ordered as (I_lon, I_lat, I_0, …, I_n) for WCS maps and (I_hpx, I_0, …, I_n) for HEALPix maps. Pixel indices can be either float or integer type. Float indices will be rounded to the nearest integer.
- vals
ndarray
Values vector.
-
slice_by_idx
(slices)¶ Slice sub map from map object.
-
sum_over_axes
(axes_names=None, keepdims=True, weights=None)¶ To sum map values over all non-spatial axes.
- Parameters
- keepdimsbool, optional
If this is set to true, the axes which are summed over are left in the map with a single bin
- axes_names: list of str
Names of MapAxis to reduce over. If None, all will summed over
- weights
Map
Weights to be applied. The Map should have the same geometry.
- Returns
- map_out
Map
Map with non-spatial axes summed over
- map_out
-
to_cube
(axes)¶ Append non-spatial axes to create a higher-dimensional Map.
This will result in a Map with a new geometry with N+M dimensions where N is the number of current dimensions and M is the number of axes in the list. The data is reshaped onto the new geometry
- Parameters
- axeslist
Axes that will be appended to this Map. The axes should have only one bin
- Returns
- map
WcsNDMap
new map
- map
-
to_hdu
(hdu='SKYMAP', hdu_bands=None, sparse=False)[source]¶ Make a FITS HDU from this map.
- Parameters
- hdustr
The HDU extension name.
- hdu_bandsstr
The HDU extension name for BANDS table.
- sparsebool
Set INDXSCHM to SPARSE and sparsify the map by only writing pixels with non-zero amplitude.
- Returns
- hdu
BinTableHDU
orImageHDU
HDU containing the map data.
- hdu
-
to_hdulist
(hdu=None, hdu_bands=None, sparse=False, format='gadf')[source]¶ Convert to
HDUList
.- Parameters
- hdustr
Name or index of the HDU with the map data.
- hdu_bandsstr
Name or index of the HDU with the BANDS table.
- sparsebool
Sparsify the map by only writing pixels with non-zero amplitude.
- format{‘gadf’, ‘fgst-ccube’,’fgst-template’}
FITS format convention.
- Returns
- hdu_list
HDUList
- hdu_list
-
to_unit
(unit)¶ Convert map to different unit
- Parameters
- unit
Unit
or str New unit
- unit
- Returns
- map
Map
Map with new unit and converted data
- map
-
abstract
upsample
(factor, order=0, preserve_counts=True, axis=None)¶ Upsample the spatial dimension by a given factor.
- Parameters
- factorint
Upsampling factor.
- orderint
Order of the interpolation used for upsampling.
- preserve_countsbool
Preserve the integral over each bin. This should be true if the map is an integral quantity (e.g. counts) and false if the map is a differential quantity (e.g. intensity).
- axisstr
Which axis to upsample. By default spatial axes are upsampled.
- Returns
- map
Map
Upsampled map.
- map
-
write
(filename, overwrite=False, **kwargs)¶ Write to a FITS file.
- Parameters
- filenamestr
Output file name.
- overwritebool
Overwrite existing file?
- hdustr
Set the name of the image extension. By default this will be set to SKYMAP (for BINTABLE HDU) or PRIMARY (for IMAGE HDU).
- hdu_bandsstr
Set the name of the bands table extension. By default this will be set to BANDS.
- formatstr, optional
FITS format convention. By default files will be written to the gamma-astro-data-formats (GADF) format. This option can be used to write files that are compliant with format conventions required by specific software (e.g. the Fermi Science Tools). The following formats are supported:
“gadf” (default)
“fgst-ccube”
“fgst-ltcube”
“fgst-bexpcube”
“fgst-srcmap”
“fgst-template”
“fgst-srcmap-sparse”
“galprop”
“galprop2”
- sparsebool
Sparsify the map by dropping pixels with zero amplitude. This option is only compatible with the ‘gadf’ format.
-