This module provides functions for plotting cartesian and polar plots. This class provides a basic plotting capability, with a minimum number of lines. These are all wrapper functions, based on existing functions in other Python classes. Provision is made for combinations of linear and log scales, as well as polar plots for two-dimensional graphs. The Plotter class can save files to disk in a number of formats.
See the __main__ function for examples of use.
This package was partly developed to provide additional material in support of students and readers of the book Electro-Optical System Analysis and Design: A Radiometry Perspective, Cornelius J. Willers, ISBN 9780819495693, SPIE Monograph Volume PM236, SPIE Press, 2013. http://spie.org/x648.html?product_id=2021423&origin_id=x646
Encapsulates a plotting environment, optimized for compact code.
This class provides a wrapper around Matplotlib to provide a plotting environment specialised towards typical pyradi visualisation. These functions were developed to provide sophisticated plots by entering the various plot options on a few lines, instead of typing many commands.
Provision is made for plots containing subplots (i.e., multiple plots on the same figure), linear scale and log scale plots, images, and cartesian, 3-D and polar plots.
Set a sequence of default colour styles of appropriate length.
The constructor provides a sequence with length 14 pre-defined plot styles. The user can define a new sequence if required. This function modulus-folds either sequence, in case longer sequences are required.
Colours can be one of the basic colours: [‘b’, ‘g’, ‘r’, ‘c’, ‘m’, ‘y’, ‘k’] or it can be a gray shade float value between 0 and 1, such as ‘0.75’, or it can be in hex format ‘#eeefff’ or it can be one of the legal html colours. See http://html-color-codes.info/ and http://www.computerhope.com/htmcolor.htm.
- Args:
plotCol ([strings]): User-supplied listof plotting styles(can be empty []).n (int): Length of required sequence.- Returns:
A list with sequence of plot styles, of required length.- Raises:
No exception is raised.
Returns a handle to the current figure
Returns a handle to the subplot, as requested per subplot number. Subplot numbers range from 1 upwards.
- Args:
subplotNumer (int) : number of the subplot- Returns:
A handle to the requested subplot or None if not found.- Raises:
No exception is raised.
Set the sub-figure title and axes labels (cartesian plots only).
Plot data on logarithmic scales for abscissa and ordinates.
Given an existing figure, this function plots in a specified subplot position. The function arguments are described below in some detail. Note that the y-values or ordinates can be more than one column, each column representing a different line in the plot. This is convenient if large arrays of data must be plotted. If more than one column is present, the label argument can contain the legend labels for each of the columns/lines. The pltaxis argument defines the min/max scale values for the x and y axes.
- Args:
plotnum (int): subplot number, 1-based indexx (np.array[N,] or [N,1]): abscissay (np.array[N,] or [N,M]): ordinates - could be M columnsptitle (string): plot title (optional)xlabel (string): x-axis label (optional)ylabel (string): y-axis label (optional)plotCol ([strings]): plot colour and line style, list with M entries, use default if [] (optional)linewidths ([float]): plot line width in points, list with M entries, use default if None (optional)label ([strings]): legend label for ordinate, list with M entries (optional)legendAlpha (float): transparency for legend box (optional)pltaxis ([xmin, xmax, ymin,ymax]): scale for x,y axes. Let Matplotlib decide if None. (optional)maxNX (int): draw maxNX+1 tick labels on x axis (optional)maxNY (int): draw maxNY+1 tick labels on y axis (optional)linestyle (string): linestyle for this plot (optional)powerLimits[float]: scientific tick label power limits [x-low, x-high, y-low, y-high] (optional) (optional)titlefsize (int): title font size, default 12pt (optional)xylabelfsize (int): x-axis, y-axis label font size, default 12pt (optional)xytickfsize (int): x-axis, y-axis tick font size, default 10pt (optional)labelfsize (int): label/legend font size, default 10pt (optional)xScientific (bool): use scientific notation on x axis (optional)yScientific (bool): use scientific notation on y axis (optional)drawGrid (bool): draw the grid on the plot (optional)yInvert (bool): invert the y-axis (optional)xInvert (bool): invert the x-axis (optional)xIsDate (bool): convert the datetime x-values to dates (optional)xTicks ({tick:label}): dict of x-axis tick locations and associated labels (optional)xtickRotation (float) x-axis tick label rotation angle (optional)markers ([string]) markers to be used for plotting data points (optional)markevery (int | (startind, stride)) subsample when using markers (optional)zorders ([int]) list of zorder for drawing sequence, highest is last (optional)clip_on (bool) clips objects to drawing axes (optional)- Returns:
the axis object for the plot- Raises:
No exception is raised.
XY colour mesh plot for (xvals, yvals, zvals) input sets.
Given an existing figure, this function plots in a specified subplot position. Only one mesh is drawn at a time. Future meshes in the same subplot will cover any previous meshes.
The mesh grid is defined in (x,y), while the height of the mesh is the z value.
The data set must have three two dimensional arrays, each for x, y, and z. The data in x, y, and z arrays must have matching data points. The x and y arrays each define the grid in terms of x and y values, i.e., the x array contains the x values for the data set, while the y array contains the y values. The z array contains the z values for the corresponding x and y values in the mesh.
Use wireframe=True to obtain a wireframe plot.
Use surface=True to obtain a surface plot with fill colours.
Z-values can be plotted on a log scale, in which case the colourbar is adjusted to show true values, but on the nonlinear scale.
The xvals and yvals vectors may have non-constant grid-intervals, i.e., they do not have to be on regular intervals, but z array must correspond to the (x,y) grid.
- Args:
plotnum (int): subplot number, 1-based indexxvals (np.array[N,M]): array of x values, corresponding to (x,y) gridyvals (np.array[N,M]): array of y values, corresponding to (x,y) gridzvals (np.array[N,M]): array of z values, corresponding to (x,y) gridptitle (string): plot title (optional)xlabel (string): x axis label (optional)ylabel (string): y axis label (optional)zlabel (string): z axis label (optional)rstride (int): mesh line row (y axis) stride, every rstride value along y axis (optional)cstride (int): mesh line column (x axis) stride, every cstride value along x axis (optional)linewidth (float): mesh line width in points (optional)plotCol ([strings]): fill colour, list with M=1 entries, use default if None (optional)edgeCol ([strings]): mesh line colour , list with M=1 entries, use default if None (optional)pltaxis ([xmin, xmax, ymin, ymax]): scale for x,y axes. z scale is not settable. Let Matplotlib decide if None (optional)maxNX (int): draw maxNX+1 tick labels on x axis (optional)maxNY (int): draw maxNY+1 tick labels on y axis (optional)maxNZ (int): draw maxNY+1 tick labels on z axis (optional)xScientific (bool): use scientific notation on x axis (optional)yScientific (bool): use scientific notation on y axis (optional)zScientific (bool): use scientific notation on z-axis (optional)powerLimits[float]: scientific tick label power limits [x-neg, x-pos, y-neg, y-pos, z-neg, z-pos] (optional)titlefsize (int): title font size, default 12pt (optional)xylabelfsize (int): x-axis, y-axis, z-axis label font size, default 12pt (optional)xytickfsize (int): x-axis, y-axis, z-axis tick font size, default 10pt (optional)wireframe (bool): If True, do a wireframe plot, (optional)surface (bool): If True, do a surface plot, (optional)cmap (cm): color map for the mesh (optional)cbarshow (bool): if true, the show a color bar (optional)cbarorientation (string): ‘vertical’ (right) or ‘horizontal’ (below) (optional)cbarcustomticks zip([z values/float],[tick labels/string]): define custom colourbar ticks locations for given z values(optional)cbarfontsize (int): font size for color bar (optional)drawGrid (bool): draw the grid on the plot (optional)xInvert (bool): invert the x-axis. Flip the x-axis left-right (optional)yInvert (bool): invert the y-axis. Flip the y-axis left-right (optional)zInvert (bool): invert the z-axis. Flip the z-axis up-down (optional)logScale (bool): do Z values on log scale, recompute colourbar vals (optional)alpha (float): surface transparency (optional)alphawire (float): mesh transparency (optional)azim (float): graph view azimuth angle [degrees] (optional)elev (float): graph view evelation angle [degrees] (optional)zorder ([int]) list of zorder for drawing sequence, highest is last (optional)clip_on (bool) clips objects to drawing axes (optional)- Returns:
the axis object for the plot- Raises:
No exception is raised.
XY colour mesh countour plot for (xvals, yvals, zvals) input sets.
The data values must be given on a fixed mesh grid of three-dimensional $(x,y,z)$ array input sets. The mesh grid is defined in $(x,y)$, while the height of the mesh is the $z$ value.
Given an existing figure, this function plots in a specified subplot position. Only one contour plot is drawn at a time. Future contours in the same subplot will cover any previous contours.
The data set must have three two dimensional arrays, each for x, y, and z. The data in x, y, and z arrays must have matching data points. The x and y arrays each define the grid in terms of x and y values, i.e., the x array contains the x values for the data set, while the y array contains the y values. The z array contains the z values for the corresponding x and y values in the contour mesh.
Z-values can be plotted on a log scale, in which case the colourbar is adjusted to show true values, but on the nonlinear scale.
The current version only saves png files, since there appears to be a problem saving eps files.
The xvals and yvals vectors may have non-constant grid-intervals, i.e., they do not have to be on regular intervals.
- Args:
plotnum (int): subplot number, 1-based indexxvals (np.array[N,M]): array of x valuesyvals (np.array[N,M]): array of y valueszvals (np.array[N,M]): values on a (x,y) gridlevels (int or [float]): number of contour levels or a list of levels (optional)ptitle (string): plot title (optional)xlabel (string): x axis label (optional)ylabel (string): y axis label (optional)shading (string): type of shading, ‘flat’ | ‘gouraud’ (optional)plotCol ([strings]): plot colour and line style, list with M entries, use default if [] (optional)pltaxis ([xmin, xmax, ymin,ymax]): scale for x,y axes. Let Matplotlib decide if None. (optional)maxNX (int): draw maxNX+1 tick labels on x axis (optional)maxNY (int): draw maxNY+1 tick labels on y axis (optional)xScientific (bool): use scientific notation on x axis (optional)yScientific (bool): use scientific notation on y axis (optional)powerLimits[float]: scientific tick label power limits [x-low, x-high, y-low, y-high] (optional)titlefsize (int): title font size, default 12pt (optional)xylabelfsize (int): x-axis, y-axis label font size, default 12pt (optional)xytickfsize (int): x-axis, y-axis tick font size, default 10pt (optional)meshCmap (cm): colour map for the mesh (optional)cbarshow (bool): if true, the show a colour bar (optional)cbarorientation (string): ‘vertical’ (right) or ‘horizontal’ (below) (optional)cbarcustomticks zip([z values/float],[tick labels/string])` define custom colourbar ticks locations for given z values(optional)cbarfontsize (int): font size for colour bar (optional)drawGrid (bool): draw the grid on the plot (optional)yInvert (bool): invert the y-axis. Flip the y-axis up-down (optional)xInvert (bool): invert the x-axis. Flip the x-axis left-right (optional)contourFill (bool): fill contours with colour (optional)contourLine (bool): draw a series of contour lines (optional)logScale (bool): do Z values on log scale, recompute colourbar values (optional)negativeSolid (bool): draw negative contours in solid lines, dashed otherwise (optional)zeroContourLine (bool): draw the contour at zero (optional)contLabel (bool): label the contours with values (optional)contFmt (string): contour label c-printf format (optional)contCol (string): contour label colour, e.g., ‘k’ (optional)contFonSz (float): contour label fontsize (optional)contLinWid (float): contour line width in points (optional)zorder ([int]) list of zorder for drawing sequence, highest is last (optional)clip_on (bool) clips objects to drawing axes (optional)- Returns:
the axis object for the plot- Raises:
No exception is raised.
Low level helper function to create a subplot and plot the data as required.
This function does the actual plotting, labelling etc. It uses the plotting function provided by its user functions.
lineStyles = { ‘’: ‘_draw_nothing’, ‘ ‘: ‘_draw_nothing’, ‘None’: ‘_draw_nothing’, ‘–’: ‘_draw_dashed’, ‘-.’: ‘_draw_dash_dot’, ‘-‘: ‘_draw_solid’, ‘:’: ‘_draw_dotted’}
- Args:
plotcommand: name of a MatplotLib plotting functionplotnum (int): subplot number, 1-based indexptitle (string): plot titlexlabel (string): x axis labelylabel (string): y axis labelx (np.array[N,] or [N,1]): abscissay (np.array[N,] or [N,M]): ordinates - could be M columnsplotCol ([strings]): plot colour and line style, list with M entries, use default if []linewidths ([float]): plot line width in points, list with M entries, use default if None (optional)label ([strings]): legend label for ordinate, list with M entrieslegendAlpha (float): transparency for legend boxpltaxis ([xmin, xmax, ymin,ymax]): scale for x,y axes. Let Matplotlib decide if None.maxNX (int): draw maxNX+1 tick labels on x axismaxNY (int): draw maxNY+1 tick labels on y axislinestyle (string): linestyle for this plot (optional)powerLimits[float]: scientific tick label power limits [x-low, x-high, y-low, y-high] (optional)titlefsize (int): title font size, default 12pt (optional)xylabelfsize (int): x-axis, y-axis label font size, default 12pt (optional)xytickfsize (int): x-axis, y-axis tick font size, default 10pt (optional)labelfsize (int): label/legend font size, default 10pt (optional)drawGrid (bool): draw the grid on the plot (optional)xScientific (bool): use scientific notation on x axis (optional)yScientific (bool): use scientific notation on y axis (optional)yInvert (bool): invert the y-axis (optional)xInvert (bool): invert the x-axis (optional)xIsDate (bool): convert the datetime x-values to dates (optional)xTicks ({tick:label}): dict of x-axis tick locations and associated labels (optional)xtickRotation (float) x-axis tick label rotation angle (optional)markers ([string]) markers to be used for plotting data points (optional)markevery (int | (startind, stride)) subsample when using markers (optional)zorders ([int]) list of zorder for drawing sequence, highest is last (optional)clip_on (bool) clips objects to drawing axes (optional)- Returns:
the axis object for the plot- Raises:
No exception is raised.
Returns the next entry in a sequence of default plot line colour styles in circular list. One day I want to do this with a generator....
- Args:
None- Returns:
The next plot colour in the sequence.- Raises:
No exception is raised.
Cartesian plot on linear scales for abscissa and ordinates.
Given an existing figure, this function plots in a specified subplot position. The function arguments are described below in some detail. Note that the y-values or ordinates can be more than one column, each column representing a different line in the plot. This is convenient if large arrays of data must be plotted. If more than one column is present, the label argument can contain the legend labels for each of the columns/lines. The pltaxis argument defines the min/max scale values for the x and y axes.
- Args:
plotnum (int): subplot number, 1-based indexx (np.array[N,] or [N,1]): abscissay (np.array[N,] or [N,M]): ordinates - could be M columnsptitle (string): plot title (optional)xlabel (string): x-axis label (optional)ylabel (string): y-axis label (optional)plotCol ([strings]): plot colour and line style, list with M entries, use default if [] (optional)linewidths ([float]): plot line width in points, list with M entries, use default if None (optional)label ([strings]): legend label for ordinate, list with M entries (optional)legendAlpha (float): transparency for legend box (optional)pltaxis ([xmin, xmax, ymin,ymax]): scale for x,y axes. Let Matplotlib decide if None. (optional)maxNX (int): draw maxNX+1 tick labels on x axis (optional)maxNY (int): draw maxNY+1 tick labels on y axis (optional)linestyle (string): linestyle for this plot (optional)powerLimits[float]: scientific tick label power limits [x-low, x-high, y-low, y-high] (optional)titlefsize (int): title font size, default 12pt (optional)xylabelfsize (int): x-axis, y-axis label font size, default 12pt (optional)xytickfsize (int): x-axis, y-axis tick font size, default 10pt (optional)labelfsize (int): label/legend font size, default 10pt (optional)xScientific (bool): use scientific notation on x axis (optional)yScientific (bool): use scientific notation on y axis (optional)drawGrid (bool): draw the grid on the plot (optional)yInvert (bool): invert the y-axis (optional)xInvert (bool): invert the x-axis (optional)xIsDate (bool): convert the datetime x-values to dates (optional)xTicks ({tick:label}): dict of x-axis tick locations and associated labels (optional)xtickRotation (float) x-axis tick label rotation angle (optional)markers ([string]) markers to be used for plotting data points (optional)markevery (int | (startind, stride)) subsample when using markers (optional)zorders ([int]) list of zorder for drawing sequence, highest is last (optional)clip_on (bool) clips objects to drawing axes (optional)- Returns:
the axis object for the plot- Raises:
No exception is raised.
3D plot on linear scales for x y z input sets.
Given an existing figure, this function plots in a specified subplot position. The function arguments are described below in some detail.
Note that multiple 3D data sets can be plotted simultaneously by adding additional columns to the input coordinates of the (x,y,z) arrays, each set of columns representing a different line in the plot. This is convenient if large arrays of data must be plotted. If more than one column is present, the label argument can contain the legend labels for each of the columns/lines.
- Args:
plotnum (int): subplot number, 1-based indexx (np.array[N,] or [N,M]) x coordinates of each line.y (np.array[N,] or [N,M]) y coordinates of each line.z (np.array[N,] or [N,M]) z coordinates of each line.ptitle (string): plot title (optional)xlabel (string): x-axis label (optional)ylabel (string): y-axis label (optional)zlabel (string): z axis label (optional)plotCol ([strings]): plot colour and line style, list with M entries, use default if None (optional)linewidths ([float]): plot line width in points, list with M entries, use default if None (optional)pltaxis ([xmin, xmax, ymin, ymax, zmin, zmax]) scale for x,y,z axes. Let Matplotlib decide if None. (optional)label ([strings]): legend label for ordinate, list with M entries (optional)legendAlpha (float): transparency for legend box (optional)titlefsize (int): title font size, default 12pt (optional)xylabelfsize (int): x, y, z label font size, default 12pt (optional)xInvert (bool): invert the x-axis (optional)yInvert (bool): invert the y-axis (optional)zInvert (bool): invert the z-axis (optional)scatter (bool): draw only the points, no lines (optional)markers ([string]): markers to be used for plotting data points (optional)markevery (int | (startind, stride)): subsample when using markers (optional)azim (float): graph view azimuth angle [degrees] (optional)elev (float): graph view evelation angle [degrees] (optional)zorder ([int]) list of zorder for drawing sequence, highest is last (optional)clip_on (bool) clips objects to drawing axes (optional)- Returns:
the axis object for the plot- Raises:
No exception is raised.
Creates a plot from an input array.
Given an input array with m x n dimensions, this function creates a subplot for vectors [1-n]. Vector 0 serves as the x-axis for each subplot. The slice dimension can be in columns (0) or rows (1).
- Args:
plotnum (int): subplot number, 1-based indexinarray (array): np.arrayslicedim (int): slice along columns (0) or rows (1)subtitles (list): a list of strings as subtitles for each subplotxlabel (string): x-axis label (optional)maxNX (int): draw maxNX+1 tick labels on x axis (optional)maxNY (int): draw maxNY+1 tick labels on y axis (optional)titlefsize (int): title font size, default 12pt (optional)xylabelfsize (int): x-axis, y-axis label font size, default 12pt (optional)xytickfsize (int): x-axis, y-axis tick font size, default 10pt (optional)- Returns:
Nothing- Raises:
No exception is raised.
Create a subplot and plot the data in polar coordinates (linear radial orginates only).
Given an existing figure, this function plots in a specified subplot position. The function arguments are described below in some detail. Note that the radial values or ordinates can be more than one column, each column representing a different line in the plot. This is convenient if large arrays of data must be plotted. If more than one column is present, the label argument can contain the legend labels for each of the columns/lines. The scale for the radial ordinates can be set with rscale. The number of radial grid circles can be set with rgrid - this provides a somewhat better control over the built-in radial grid in matplotlib. thetagrids defines the angular grid interval. The angular rotation direction can be set to be clockwise or counterclockwise. Likewise, the rotation offset where the plot zero angle must be, is set with zerooffset.
For some obscure reason Matplitlib version 1.13 does not plot negative values on the polar plot. We therefore force the plot by making the values positive and then highlight it as negative.
- Args:
plotnum (int): subplot number, 1-based indextheta (np.array[N,] or [N,1]): angular abscissa in radiansr (np.array[N,] or [N,M]): radial ordinates - could be M columnsptitle (string): plot title (optional)plotCol ([strings]): plot colour and line style, list with M entries, use default if None (optional)label ([strings]): legend label, list with M entries (optional)labelLocation ([x,y]): where the legend should located (optional)highlightNegative (bool): indicate if negative data must be highlighted (optional)highlightCol (string): negative highlight colour string (optional)highlightWidth (int): negative highlight line width(optional)legendAlpha (float): transparency for legend box (optional)rscale ([rmin, rmax]): radial plotting limits. use default setting if None. If rmin is negative the zero is a circle and rmin is at the centre of the graph (optional)rgrid ([rinc, numinc]): radial grid, use default is [0,5]. If rgrid is None don’t show. If rinc=0 then numinc is number of intervals. If rinc is not zero then rinc is the increment and numinc is ignored (optional)thetagrids (float): theta grid interval [degrees], if None don’t show (optional)direction (string): direction in increasing angle, ‘counterclockwise’ or ‘clockwise’ (optional)zerooffset (float): rotation offset where zero should be [rad]. Positive zero-offset rotation is counterclockwise from 3’o’clock (optional)titlefsize (int): title font size, default 12pt (optional)drawGrid (bool): draw a grid on the graph (optional)zorder ([int]) list of zorder for drawing sequence, highest is last (optional)clip_on (bool) clips objects to drawing axes (optional)- Returns:
the axis object for the plot- Raises:
No exception is raised.
3D polar surface/mesh plot for (r, theta, zvals) input sets.
Given an existing figure, this function plots in a specified subplot position.
Only one mesh is drawn at a time. Future meshes in the same subplot will cover any previous meshes.
The data in zvals must be on a grid where the theta vector correspond to the number of rows in zvals and the radial vector corresponds to the number of columns in zvals.
The r and p vectors may have non-constant grid-intervals, i.e., they do not have to be on regular intervals.
- Args:
plotnum (int): subplot number, 1-based indextheta (np.array[N,M]): array of angular values [0..2pi] corresponding to (theta,rho) grid.radial (np.array[N,M]): array of radial values corresponding to (theta,rho) grid.zvals (np.array[N,M]): array of z values corresponding to (theta,rho) grid.ptitle (string): plot title (optional)xlabel (string): x-axis label (optional)ylabel (string): y-axis label (optional)zlabel (string): z-axis label (optional)zscale ([float]): z axis [min, max] in the plot.titlefsize (int): title font size, default 12pt (optional)xylabelfsize (int): x, y, z label font size, default 12pt (optional)thetaStride (int): theta stride in input data (optional)radialstride (int): radial stride in input data (optional)meshCmap (cm): color map for the mesh (optional)linewidth (float): width of the mesh linesazim (float): graph view azimuth angle [degrees] (optional)elev (float): graph view evelation angle [degrees] (optional)zorder ([int]) list of zorder for drawing sequence, highest is last (optional)clip_on (bool) clips objects to drawing axes (optional)- Returns:
the axis object for the plot- Raises:
No exception is raised.
Polar colour contour and filled contour plot for (theta, r, zvals) input sets.
The data values must be given on a fixed mesh grid of three-dimensional (theta,rho,z) array input sets (theta is angle, and rho is radial distance). The mesh grid is defined in (theta,rho), while the height of the mesh is the z value. The (theta,rho) arrays may have non-constant grid-intervals, i.e., they do not have to be on regular intervals.
Given an existing figure, this function plots in a specified subplot position. Only one contour plot is drawn at a time. Future contours in the same subplot will cover any previous contours.
The data set must have three two dimensional arrays, each for theta, rho, and z. The data in theta, rho, and z arrays must have matching data points. The theta and rho arrays each define the grid in terms of theta and rho values, i.e., the theta array contains the angular values for the data set, while the rho array contains the radial values. The z array contains the z values for the corresponding theta and rho values in the contour mesh.
Z-values can be plotted on a log scale, in which case the colourbar is adjusted to show true values, but on the nonlinear scale.
The current version only saves png files, since there appears to be a problem saving eps files.
- Args:
plotnum (int): subplot number, 1-based indextheta (np.array[N,M]) array of angular values [0..2pi] corresponding to (theta,rho) grid.radial (np.array[N,M]) array of radial values corresponding to (theta,rho) grid.zvals (np.array[N,M]) array of z values corresponding to (theta,rho) grid.ptitle (string): plot title (optional)shading (string): ‘flat’ | ‘gouraud’ (optional)radscale ([float]): inner and outer radial scale max in the plot.titlefsize (int): title font size, default 12pt (optional)meshCmap (cm): color map for the mesh (optional)cbarshow (bool): if true, the show a color barcbarorientation (string): ‘vertical’ (right) or ‘horizontal’ (below)cbarcustomticks zip([tick locations/float],[tick labels/string]): locations in image grey levelscbarfontsize (int): font size for color barrgrid ([float]): radial grid - None, [number], [inc,max]thetagrid ([float]): angular grid - None, [inc]drawGrid (bool): draw the grid on the plot (optional)thetagridfontsize (float): font size for the angular gridradialgridfontsize (float): font size for the radial griddirection (string)= ‘counterclockwise’ or ‘clockwise’ (optional)zerooffset (float) = rotation offset where zero should be [rad] (optional)logScale (bool): do Z values on log scale, recompute colourbar valsplotCol ([strings]): plot colour and line style, list with M entries, use default if []levels (int or [float]): number of contour levels or a list of levels (optional)contourFill (bool): fill contours with colour (optional)contourLine (bool): draw a series of contour lineszeroContourLine (bool): draw the contour at zeronegativeSolid (bool): draw negative contours in solid lines, dashed otherwise (optional)contLabel (bool): label the contours with values (optional)contFmt (string): contour label c-printf format (optional)contCol (string): contour label colour, e.g., ‘k’ (optional)contFonSz (float): contour label fontsize (optional)contLinWid (float): contour line width in points (optional)zorder ([int]) list of zorder for drawing sequence, highest is last (optional)clip_on (bool) clips objects to drawing axes (optional)- Returns:
the axis object for the plot- Raises:
No exception is raised.
Resets the plot colours to start at the beginning of the cycle.
- Args:
None
- Returns:
None.- Raises:
No exception is raised.
Save the plot to a disk file, using filename, dpi specification and bounding box limits.
One of matplotlib’s design choices is a bounding box strategy which may result in a bounding box that is smaller than the size of all the objects on the page. It took a while to figure this out, but the current default values for bbox_inches and pad_inches seem to create meaningful bounding boxes. These are however larger than the true bounding box. You still need a tool such as epstools or Adobe Acrobat to trim eps files to the true bounding box.
The type of file written is picked up in the filename. Most backends support png, pdf, ps, eps and svg.
- Args:
filename (string): output filename to write plot, file extdpi (int): the resolution of the graph in dots per inchbbox_inches: see matplotlib docs for more detail.pad_inches: see matplotlib docs for more detail.useTrueType: if True, truetype fonts are used in eps/pdf files, otherwise Type3- Returns:
Nothing. Saves a file to disk.- Raises:
No exception is raised.
Plot data on logarithmic scales for abscissa and linear scale for ordinates.
Given an existing figure, this function plots in a specified subplot position. The function arguments are described below in some detail. Note that the y-values or ordinates can be more than one column, each column representing a different line in the plot. This is convenient if large arrays of data must be plotted. If more than one column is present, the label argument can contain the legend labels for each of the columns/lines. The pltaxis argument defines the min/max scale values for the x and y axes.
- Args:
plotnum (int): subplot number, 1-based indexx (np.array[N,] or [N,1]): abscissay (np.array[N,] or [N,M]): ordinates - could be M columnsptitle (string): plot title (optional)xlabel (string): x-axis label (optional)ylabel (string): y-axis label (optional)plotCol ([strings]): plot colour and line style, list with M entries, use default if [] (optional)linewidths ([float]): plot line width in points, list with M entries, use default if None (optional)label ([strings]): legend label for ordinate, list with M entries (optional)legendAlpha (float): transparency for legend box (optional)pltaxis ([xmin, xmax, ymin,ymax]): scale for x,y axes. Let Matplotlib decide if None. (optional)maxNX (int): draw maxNX+1 tick labels on x axis (optional)maxNY (int): draw maxNY+1 tick labels on y axis (optional)linestyle (string): linestyle for this plot (optional)powerLimits[float]: scientific tick label notation power limits [x-low, x-high, y-low, y-high] (optional) (optional)titlefsize (int): title font size, default 12pt (optional)xylabelfsize (int): x-axis, y-axis label font size, default 12pt (optional)xytickfsize (int): x-axis, y-axis tick font size, default 10pt (optional)labelfsize (int): label/legend font size, default 10pt (optional)xScientific (bool): use scientific notation on x axis (optional)yScientific (bool): use scientific notation on y axis (optional)drawGrid (bool): draw the grid on the plot (optional)yInvert (bool): invert the y-axis (optional)xInvert (bool): invert the x-axis (optional)xIsDate (bool): convert the datetime x-values to dates (optional)xTicks ({tick:label}): dict of x-axis tick locations and associated labels (optional)xtickRotation (float) x-axis tick label rotation angle (optional)markers ([string]) markers to be used for plotting data points (optional)markevery (int | (startind, stride)) subsample when using markers (optional)zorders ([int]) list of zorder for drawing sequence, highest is last (optional)clip_on (bool) clips objects to drawing axes (optional)- Returns:
the axis object for the plot- Raises:
No exception is raised.
Plot data on linear scales for abscissa and logarithmic scale for ordinates.
Given an existing figure, this function plots in a specified subplot position. The function arguments are described below in some detail. Note that the y-values or ordinates can be more than one column, each column representing a different line in the plot. This is convenient if large arrays of data must be plotted. If more than one column is present, the label argument can contain the legend labels for each of the columns/lines. The pltaxis argument defines the min/max scale values for the x and y axes.
- Args:
plotnum (int): subplot number, 1-based indexx (np.array[N,] or [N,1]): abscissay (np.array[N,] or [N,M]): ordinates - could be M columnsptitle (string): plot title (optional)xlabel (string): x-axis label (optional)ylabel (string): y-axis label (optional)plotCol ([strings]): plot colour and line style, list with M entries, use default if [] (optional)linewidths ([float]): plot line width in points, list with M entries, use default if None (optional)label ([strings]): legend label for ordinate, list withM entries (optional)legendAlpha (float): transparency for legend box (optional)pltaxis ([xmin, xmax, ymin,ymax]): scale for x,y axes. Let Matplotlib decide if None. (optional)maxNX (int): draw maxNX+1 tick labels on x axis (optional)maxNY (int): draw maxNY+1 tick labels on y axis (optional)linestyle (string): linestyle for this plot (optional)powerLimits[float]: scientific tick label power limits [x-low, x-high, y-low, y-high] (optional) (optional)titlefsize (int): title font size, default 12pt (optional)xylabelfsize (int): x-axis, y-axis label font size, default 12pt (optional)xytickfsize (int): x-axis, y-axis tick font size, default 10pt (optional)labelfsize (int): label/legend font size, default 10pt (optional)xScientific (bool): use scientific notation on x axis (optional)yScientific (bool): use scientific notation on y axis (optional)drawGrid (bool): draw the grid on the plot (optional)yInvert (bool): invert the y-axis (optional)xInvert (bool): invert the x-axis (optional)xIsDate (bool): convert the datetime x-values to dates (optional)xTicks ({tick:label}): dict of x-axis tick locations and associated labels (optional)xtickRotation (float) x-axis tick label rotation angle (optional)markers ([string]) markers to be used for plotting data points (optional)markevery (int | (startind, stride)) subsample when using markers (optional)zorders ([int]) list of zorder for drawing sequence, highest is last (optional)clip_on (bool) clips objects to drawing axes (optional)- Returns:
the axis object for the plot- Raises:
No exception is raised.
Creates a subplot and show the image using the colormap provided.
Filled marker user-settable values.
This class encapsulates a few variables describing a Filled marker. Default values are provided that can be overridden in user plots.
Collect marker location and types and mark subplot.
Build a list of markers at plot locations with the specified marker.
Add a marker to the list, overridding properties if necessary.
Specify location and any specific marker properties to be used. The location can be (xy,y) for cartesian plots or (theta,rad) for polars.
If no marker properties are specified, the current marker class properties will be used. If the current maker instance does not specify properties, the default marker properties will be used.
- Args:
x (float): the x/theta location for the markery (float): the y/radial location for the markermarkerfacecolor (colour): main colour for marker (optional)markerfacecoloralt (colour): alterive colour for marker (optional)markeredgecolor (colour): edge colour for marker (optional)marker (string): string to specify the marker (optional)markersize (int)): size of the marker (optional)fillstyle (string): string to define fill style (optional)- Returns:
Nothing. Creates the figure for subequent use.- Raises:
No exception is raised.
Plot the current list of markers on the given axes.
All the markers currently stored in the class will be drawn.
- Args:
ax (axes): an axes handle for the plot- Returns:
Nothing. Creates the figure for subequent use.- Raises:
No exception is raised.
This class provides a functions to assist in the optimal display of images.
Compress an image (and then inversely expand the color bar values), prior to histogram equalisation to ensure that the two keep in step, we store the compression function names as pairs, and invoke the compression function as follows: linear, log. sqrt. Note that the image is histogram equalised in all cases.
image (np.ndarray): the image to be processedselectCompressSet (int): compression selection [0,1,2] (optional)numCbarlevels (int): number of labels in the colourbar (optional)cbarformat (string): colourbar label format, e.g., ‘10.3f’, ‘.5e’ (optional)
Reprojects a 3D numpy array into a polar coordinate system, relative to some origin.
This function reprojects an image from cartesial to polar coordinates. The origin of the new coordinate system defaults to the center of the image, unless the user supplies a new origin.
The data format can be data.shape = (rows, cols, frames) or data.shape = (frames, rows, cols), the format of which is indicated by the framesFirst parameter.
original code by Joe Kington https://stackoverflow.com/questions/3798333/image-information-along-a-polar-coordinate-system
Uses ‘with’ statement to create a plot and save to file on exit.
Use as follows:
x=np.linspace(-3,3,20)
with savePlot(1,saveName=['testwith.png','testwith.eps']) as p:
p.plot(1,x,x*x)
where the savePlot parameters are exactly the same as Plotter, except that a new named parameter saveName is now present. If saveName is not None, the list of filenames is used to save files of the plot (any number of names/types)
A full implementation of Dave Green’s “cubehelix” for Matplotlib. Based on the FORTRAN 77 code provided in D.A. Green, 2011, BASI, 39, 289.
http://adsabs.harvard.edu/abs/2011arXiv1108.5083G http://www.astron-soc.in/bulletin/11June/289392011.pdf
User can adjust all parameters of the cubehelix algorithm. This enables much greater flexibility in choosing color maps, while always ensuring the color map scales in intensity from black to white. A few simple examples:
Default color map settings produce the standard “cubehelix”.
Create color map in only blues by setting rot=0 and start=0.
Create reverse (white to black) backwards through the rainbow once by setting rot=1 and reverse=True.
Example: >>> import cubehelix >>> cx = cubehelix.cmap(start=0., rot=-0.5) >>> plot(x,cmap=cx)
Revisions 2014-04 (@jradavenport) Ported from IDL version
source https://github.com/jradavenport/cubehelix
Licence Copyright (c) 2014, James R. A. Davenport and contributors All rights reserved.
Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met:
Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer.
Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution.
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS “AS IS” AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.