U Cv fR'@sddlmZddlZddlZddlmZmZmZmZm Z m Z m Z m Z m Z ddlZddlZddlmZddlmZddlmZddlmZddlmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,er~dd l-m.Z.dd l/m0Z0m1Z1dd l2m3Z3m4Z4dd l5m6Z6dd l7m8Z8ddl9m:Z:ddl;mm?Z?ddl@mAZAddlBmCZCmDZDmEZEmFZFmGZGddlmHZHddddddddZIddddddddZJdd dd!d"d#d$d%ZKddddddd&dddd'd(dd)d*d*d+ d,d-ZLe dddddddddddddddddd.d.d/dd*d0d0d1d2d3d(dddd4d4d5d5d6d6d6d6d7d7d*d8d9d:d;ZMe ddddddddddddddddd.d.d<d*d=dd1d2d3d(dddd4d4d5d5d6d6d6d6d7d7d*d>d?d@d;ZMe ddddddddddddddddd.d.dAd*dd=d1d2d3d(dddd4d4d5d5d6d6d6d6d7d7d*d>d?dBd;ZMdddddddddddddddddd.d.d/dd*ddd1d2d3d(dddd4d4d5d5d6d6d6d6d7d7d*dCd9dDd;ZMe dEdddddFdd*dGdHdHd0d0d*d8dI dJdKZNe dEddddLdd*dGdHdHd=dd*d>dI dMdKZNe dEddddNdd*dGdHdHdd=d*d>dI dOdKZNdEdddddFdd*dGdHdHddd*dCdI dPdKZNdddddddddddddQ dd*d1d3d2d(d4d4d5d5d6d6d6d6d*dRdSdTdUZOdVdWZPdXdYddZd[d0d\d]d^ZQe ddddddddddddddd.d.ddd.d.dddddddddddddd_!dd*ddddd`ddd1d3d3d(d0d0d'd4d4d4d4dXd7d)d5d5d6d6d6d6dad3d3dbdcd6ddde$dfdgZRe dddddddddddddd.d.ddd.d.ddddddddddddddh dd*ddddd`ddd1d3d3d(dd=d'd4d4d4d4dXd7d)d5d5d6d6d6d6dad3d3dbdcd6d>de$didgZRe dddddddddddddd.d.ddd.d.ddddddddddddddj dd*ddddd`ddd1d3d3d(d=dd'd4d4d4d4dXd7d)d5d5d6d6d6d6dad3d3dbdcd6d>de$dkdgZRePddldld[dXdddmdndgZRdodpZSe ddddddddd.d.dd.dddddqdddddddddddddddrdddd1d3d3d(d0d0d'd4d4d4d7d3d3dad3d7dcd6dsd)d(d)d5d5d6d6d6d6dbd*dtdu"dvdwZTe dddddddd.d.dd.dddddqdddddddddddddddxdddd1d3d2d(dd=d'd4d4d4d7d3d3dad3d7dcd6dsd)d(d)d5d5d6d6d6d6dbd*d>du"dydwZTe dddddddd.d.dd.dddddqdddddddddddddddzdddd1d3d2d(d=dd'd4d4d4d7d3d3dad3d7dcd6dsd)d(d)d5d5d6d6d6d6dbd*d>du"d{dwZTeSd|d|d}d[d*dtd~ddwZTe ddddddddd.d.dd.dddddqdddddddddddddddrdddd1d3d3d(d0d0d'd4d4d4d7d3d3dad3d7dcd6dsd)d(d)d5d5d6d6d6d6dbd*ddu"ddZUe dddddddd.d.dd.dddddqdddddddddddddddxdddd1d3d2d(dd=d'd4d4d4d7d3d3dad3d7dcd6dsd)d(d)d5d5d6d6d6d6dbd*d>du"ddZUe dddddddd.d.dd.dddddqdddddddddddddddzdddd1d3d2d(d=dd'd4d4d4d7d3d3dad3d7dcd6dsd)d(d)d5d5d6d6d6d6dbd*d>du"ddZUeSd|d|d|d[d*dd~ddZUe ddddddddd.d.dd.dddddqdddddddddddddddrdddd1d3d2d(d0d0d'd4d4d4d7d3d3dad3d7dcd6dsd)d(d)d5d5d6d6d6d6dbd*ddu"ddZVe dddddddd.d.dd.dddddqdddddddddddddddxdddd1d3d2d(dd=d'd4d4d4d7d3d3dad3d7dcd6dsd)d(d)d5d5d6d6d6d6dbd*d>du"ddZVe dddddddd.d.dd.dddddqdddddddddddddddzdddd1d3d2d(d=dd'd4d4d4d7d3d3dad3d7dcd6dsd)d(d)d5d5d6d6d6d6dbd*d>du"ddZVeSd|d|d|d[d*dd~ddZVe ddddddddd.d.dd.dddddqdddddddddddddddrdddd1d3d2d(d0d0d'd4d4d4d7d3d3dad3d7dcd6dsd)d(d)d5d5d6d6d6d6dbd*ddu"ddZWe dddddddd.d.dd.dddddqdddddddddddddddxdddd1d3d2d(dd=d'd4d4d4d7d3d3dad3d7dcd6dsd)d(d)d5d5d6d6d6d6dbd*d>du"ddZWe dddddddd.d.dd.dddddqdddddddddddddddzdddd1d3d2d(d=dd'd4d4d4d7d3d3dad3d7dcd6dsd)d(d)d5d5d6d6d6d6dbd*d>du"ddZWeSdd|d|d|d[ddd*ddddZWe ddddddddd.d.dd.dddddqdddddddddddddddrdddd1d3d2d(d0d0d'd4d4d4d7d3d3dad3d7dcd6dsd)d(d)d5d5d6d6d6d6dbd*ddu"ddZXe dddddddd.d.dd.dddddqdddddddddddddddxdddd1d3d2d(dd=d'd4d4d4d7d3d3dad3d7dcd6dsd)d(d)d5d5d6d6d6d6dbd*d>du"ddZXe dddddddd.d.dd.dddddqdddddddddddddddzdddd1d3d2d(d=dd'd4d4d4d7d3d3dad3d7dcd6dsd)d(d)d5d5d6d6d6d6dbd*d>du"ddZXeSd|d|d|d[d*dd~ddZXdS)) annotationsN) TYPE_CHECKINGAnyCallableHashableIterableLiteralMutableMappingcastoverload)Version) broadcast)concat)_easy_facetgrid)_LINEWIDTH_RANGE_MARKERSIZE_RANGE _add_colorbar _add_legend_assert_valid_xy_determine_guide_ensure_plottable_infer_interval_breaks_infer_xy_labels _Normalize_process_cmap_cbar_kwargs_rescale_imshow_rgb_resolve_intervals_1dplot_resolve_intervals_2dplot _update_axesget_axisimport_matplotlib_pyplotlabel_from_attrs)Axes)PathCollectionQuadMesh)Colormap Normalize) BarContainer)QuadContourSet) AxesImage)Line3DPoly3DCollection) ArrayLike) DataArray) AspectOptions ExtendOptionsHueStyleOptions ScaleOptions T_DataArray) FacetGridr-Hashable | Nonez2tuple[DataArray, DataArray, DataArray | None, str])darrayxyhuereturncCsdt|j}|dk r"|dk r"td|dk r6t||d|dk rJt||d|dkrd}d}d}|dk rt||}|} n,|dk r|}||} n|jd} || }|} n|dkr|dkr|dkrtd|dkr|dk rt||dt|||d \} }|| }|jdkrn||jkrZ|j|dkr*dnd} |j| } |j| |d d } |j| |d d }ntd t|jn$|| j\}||j\}|||} nt|||d \}}||} | jdkr ||jkr |j|dkrdnd} |j| } |j| |d d }| j| |d d } ntd t|jn$||j\}||j\}|||}t ||}||}|| ||fS) Nz2Cannot specify both x and y kwargs for line plots.r6r7rz1For 2D inputs, please specify either hue, x or y.r8)r5r6r7F)transpose_coordsz3For 2D inputs, hue must be a dimension i.e. one of ) lendims ValueErrorrrndimindex transposereprr!)r5r6r7r8ndimsZhuenamehuepltZhuelabelxpltypltdimZxnameZ otherindexZotherdimZxdimZhuedimZynameZydimrI>/tmp/pip-unpacked-wheel-h316xyqg/xarray/plot/dataarray_plot.py_infer_line_dataBsx                   rKr6r8sizezMutableMapping[str, Hashable]z Iterable[str])r5 dims_plot default_guessr9csxdd|Dtfdd|jD}t||D] \}}||ddkr4|||<q4|D]\}}t|||q^|S)a Guess what dims to plot if some of the values in dims_plot are None which happens when the user has not defined all available ways of visualizing the data. Parameters ---------- darray : DataArray The DataArray to check. dims_plot : T_DimsPlot Dims defined by the user to plot. default_guess : Iterable[str], optional Default values and order to retrieve dims if values in dims_plot is missing, default: ("x", "hue", "size"). cSsi|]\}}|dk r||qSNrI.0kvrIrIrJ sz$_infer_plot_dims..c3s|]}|kr|VqdSrP)valuesrRrTZdims_plot_existrIrJ s z#_infer_plot_dims..N)itemstupler>zipgetr)r5rNrOZ dims_availrSrTrIrXrJ_infer_plot_dimss r^r2z None | strzdict[str, T_DataArray])r5rN plotfunc_namer9cst|}jdkrg}tjtjrdD]T}||d}|dk r,|jkr,tj |di}t |g|d| ||q,j d |j jdtd}|fdd |Dtt|t|}|S) Nr:)zr6)rH.)Z _stacked_dim)r7cs"i|]\}}|dk r||qSrPrIrQr5rIrJrUsz%_infer_line_data2..).)r^r@np issubdtypedtypeZfloatingr]r>nanZiselrappendrBstackdictupdaterZr\keysr rV)r5rNr_Zdims_TrTrHZ darray_nanoutrIrbrJ_infer_line_data2s     rm)rowcolcol_wrapaxr8 subplot_kws int | None Axes | Nonedict[str, Any] | Noner) r5rnrorprqr8rrkwargsr9c Ks|}t|j}||||||t|} | dksN|jdkrVtd| dkr|sf|r||d<||d<||d<||d<| dkrt} ||d <q| d kr|rt} ||d <qt } ||d<n|s|s|rt d t } ||d <| |f|S) a Default plot of DataArray using :py:mod:`matplotlib:matplotlib.pyplot`. Calls xarray plotting function based on the dimensions of the squeezed DataArray. =============== =========================== Dimensions Plotting function =============== =========================== 1 :py:func:`xarray.plot.line` 2 :py:func:`xarray.plot.pcolormesh` Anything else :py:func:`xarray.plot.hist` =============== =========================== Parameters ---------- darray : DataArray row : Hashable or None, optional If passed, make row faceted plots on this dimension name. col : Hashable or None, optional If passed, make column faceted plots on this dimension name. col_wrap : int or None, optional Use together with ``col`` to wrap faceted plots. ax : matplotlib axes object, optional Axes on which to plot. By default, use the current axes. Mutually exclusive with ``size``, ``figsize`` and facets. hue : Hashable or None, optional If passed, make faceted line plots with hue on this dimension name. subplot_kws : dict, optional Dictionary of keyword arguments for Matplotlib subplots (see :py:meth:`matplotlib:matplotlib.figure.Figure.add_subplot`). **kwargs : optional Additional keyword arguments for Matplotlib. See Also -------- xarray.DataArray.squeeze rNo numeric data to plot.)r:rrrnrorpr:r8rxzdOnly 1d and 2d plots are supported for facets in xarray. See the package `Seaborn` for more options.rq) ZsqueezeZcomputesetr>discardr=rM TypeErrorline pcolormeshr?hist) r5rnrorprqr8rrrvZ plot_dimsrDplotfuncrIrIrJplots<1         rT)rnrofigsizeaspectrMrqr8r6r7 xincrease yincreasexscaleyscalexticksyticksxlimylim add_legend_labelsNoneIterable[float] | Noner. float | None bool | Noner1ArrayLike | Noneboolz list[Line3D])r5argsrnrorrrMrqr8r6r7rrrrrrrrrrrvr9cOsdSrPrIr5rnrorrrMrqr8r6r7rrrrrrrrrrrrvrIrIrJr|;sr|)rorrrMrqr8r6r7rrrrrrrrrrrzFacetGrid[DataArray])rrnrorrrMrqr8r6r7rrrrrrrrrrrvr9cOsdSrPrIrrIrIrJr|Ws)rnrrrMrqr8r6r7rrrrrrrrrrcOsdSrPrIrrIrIrJr|ssz#list[Line3D] | FacetGrid[DataArray]c$ Os|s|rBt}||d|dt|tfddi|St|j}|dks^|jdkrft d|dkr~t dj |d |d kr|d d }n d |kst t ||||}t||| |\}}}}t|||\}}}}}t||d } t||d }!t|||j||f||}"|rT| d k r2|| |!d k rF||!|||jdkr|r|d k stt |j|"t||dt|jtjr|D]}#|# d|#!dqt"|| | | | |||| |"S)a Line plot of DataArray values. Wraps :py:func:`matplotlib:matplotlib.pyplot.plot`. Parameters ---------- darray : DataArray Either 1D or 2D. If 2D, one of ``hue``, ``x`` or ``y`` must be provided. row : Hashable, optional If passed, make row faceted plots on this dimension name. col : Hashable, optional If passed, make column faceted plots on this dimension name. figsize : tuple, optional A tuple (width, height) of the figure in inches. Mutually exclusive with ``size`` and ``ax``. aspect : "auto", "equal", scalar or None, optional Aspect ratio of plot, so that ``aspect * size`` gives the *width* in inches. Only used if a ``size`` is provided. size : scalar, optional If provided, create a new figure for the plot with the given size: *height* (in inches) of each plot. See also: ``aspect``. ax : matplotlib axes object, optional Axes on which to plot. By default, the current is used. Mutually exclusive with ``size`` and ``figsize``. hue : Hashable, optional Dimension or coordinate for which you want multiple lines plotted. If plotting against a 2D coordinate, ``hue`` must be a dimension. x, y : Hashable, optional Dimension, coordinate or multi-index level for *x*, *y* axis. Only one of these may be specified. The other will be used for values from the DataArray on which this plot method is called. xincrease : bool or None, optional Should the values on the *x* axis be increasing from left to right? if ``None``, use the default for the Matplotlib function. yincrease : bool or None, optional Should the values on the *y* axis be increasing from top to bottom? if ``None``, use the default for the Matplotlib function. xscale, yscale : {'linear', 'symlog', 'log', 'logit'}, optional Specifies scaling for the *x*- and *y*-axis, respectively. xticks, yticks : array-like, optional Specify tick locations for *x*- and *y*-axis. xlim, ylim : array-like, optional Specify *x*- and *y*-axis limits. add_legend : bool, default: True Add legend with *y* axis coordinates (2D inputs only). *args, **kwargs : optional Additional arguments to :py:func:`matplotlib:matplotlib.pyplot.plot`. Returns ------- primitive : list of Line3D or FacetGrid When either col or row is given, returns a FacetGrid, otherwise a list of matplotlib Line3D objects. rvr5kindr|rrwrxzZLine plots are for 1- or 2-dimensional DataArrays. Passed DataArray has {ndims} dimensions)rDrIrextraNZhandleslabelstitleright)#localscopyrjpoprr|r=r>rMr{r?formatAssertionErrorrrKrto_numpyr!rr set_xlabel set_ylabel set_title_title_for_slicer@legendlistrcrdre datetime64get_xticklabels set_rotationset_har)$r5rnrorrrMrqr8r6r7rrrrrrrrrrrrvallargsrDrFrGrEZ hue_labelZxplt_valZyplt_valZx_suffixZy_suffixZxlabelZylabel primitivexlabelsrIrIrJr|sVQ             pre)where drawstyledsrnrozLiteral[('pre', 'post', 'mid')]z str | None) r5rrrrrnrorvr9cOsdSrPrIr5rrrrnrorrvrIrIrJstep"s r)rrrrocOsdSrPrIrrIrIrJr0s )rrrrncOsdSrPrIrrIrIrJr>s cOsb|dkrtd|dk r.|dkr&|}ntd|dkr:d}d||}t|f||||d|S)a Step plot of DataArray values. Similar to :py:func:`matplotlib:matplotlib.pyplot.step`. Parameters ---------- where : {'pre', 'post', 'mid'}, default: 'pre' Define where the steps should be placed: - ``'pre'``: The y value is continued constantly to the left from every *x* position, i.e. the interval ``(x[i-1], x[i]]`` has the value ``y[i]``. - ``'post'``: The y value is continued constantly to the right from every *x* position, i.e. the interval ``[x[i], x[i+1])`` has the value ``y[i]``. - ``'mid'``: Steps occur half-way between the *x* positions. Note that this parameter is ignored if one coordinate consists of :py:class:`pandas.Interval` values, e.g. as a result of :py:func:`xarray.Dataset.groupby_bins`. In this case, the actual boundaries of the interval are used. drawstyle, ds : str or None, optional Additional drawstyle. Only use one of drawstyle and ds. row : Hashable, optional If passed, make row faceted plots on this dimension name. col : Hashable, optional If passed, make column faceted plots on this dimension name. *args, **kwargs : optional Additional arguments for :py:func:`xarray.plot.line`. Returns ------- primitive : list of Line3D or FacetGrid When either col or row is given, returns a FacetGrid, otherwise a list of matplotlib Line3D objects. >rmidpostz7'where' argument to step must be 'pre', 'post' or 'mid'Nz'ds and drawstyle are mutually exclusiver;zsteps-)rrorn)r?r{r|rrIrIrJrLs/ ) rrMrrqrrrrrrrrz+tuple[np.ndarray, np.ndarray, BarContainer])r5rrrMrrqrrrrrrrrrvr9c  Ost| dkst|jdks$|jdkr,tdt||||}t|}|t |}|j |f|}| | |t|t|||||| | | | |S)a Histogram of DataArray. Wraps :py:func:`matplotlib:matplotlib.pyplot.hist`. Plots *N*-dimensional arrays by first flattening the array. Parameters ---------- darray : DataArray Can have any number of dimensions. figsize : Iterable of float, optional A tuple (width, height) of the figure in inches. Mutually exclusive with ``size`` and ``ax``. aspect : "auto", "equal", scalar or None, optional Aspect ratio of plot, so that ``aspect * size`` gives the *width* in inches. Only used if a ``size`` is provided. size : scalar, optional If provided, create a new figure for the plot with the given size: *height* (in inches) of each plot. See also: ``aspect``. ax : matplotlib axes object, optional Axes on which to plot. By default, use the current axes. Mutually exclusive with ``size`` and ``figsize``. xincrease : bool or None, optional Should the values on the *x* axis be increasing from left to right? if ``None``, use the default for the Matplotlib function. yincrease : bool or None, optional Should the values on the *y* axis be increasing from top to bottom? if ``None``, use the default for the Matplotlib function. xscale, yscale : {'linear', 'symlog', 'log', 'logit'}, optional Specifies scaling for the *x*- and *y*-axis, respectively. xticks, yticks : array-like, optional Specify tick locations for *x*- and *y*-axis. xlim, ylim : array-like, optional Specify *x*- and *y*-axis limits. **kwargs : optional Additional keyword arguments to :py:func:`matplotlib:matplotlib.pyplot.hist`. rrw)r=rr@rMr{rrcravelrpdZnotnullr~rrrr!r)r5rrMrrqrrrrrrrrrrvZno_nanrrIrIrJr~s8r~c'sd}jd|_tjdddddddddddddddddddddddddddddddddddd!dd d d d d d d d d d d d ddddddddddddddddddddddd d$fdd}|`|S)z'Decorator for common 1d plotting logic.aW Parameters ---------- darray : DataArray Must be 2 dimensional, unless creating faceted plots. x : Hashable or None, optional Coordinate for x axis. If None use darray.dims[1]. y : Hashable or None, optional Coordinate for y axis. If None use darray.dims[0]. z : Hashable or None, optional If specified plot 3D and use this coordinate for *z* axis. hue : Hashable or None, optional Dimension or coordinate for which you want multiple lines plotted. hue_style: {'discrete', 'continuous'} or None, optional How to use the ``hue`` variable: - ``'continuous'`` -- continuous color scale (default for numeric ``hue`` variables) - ``'discrete'`` -- a color for each unique value, using the default color cycle (default for non-numeric ``hue`` variables) markersize: Hashable or None, optional scatter only. Variable by which to vary size of scattered points. linewidth: Hashable or None, optional Variable by which to vary linewidth. row : Hashable, optional If passed, make row faceted plots on this dimension name. col : Hashable, optional If passed, make column faceted plots on this dimension name. col_wrap : int, optional Use together with ``col`` to wrap faceted plots ax : matplotlib axes object, optional If None, uses the current axis. Not applicable when using facets. figsize : Iterable[float] or None, optional A tuple (width, height) of the figure in inches. Mutually exclusive with ``size`` and ``ax``. size : scalar, optional If provided, create a new figure for the plot with the given size. Height (in inches) of each plot. See also: ``aspect``. aspect : "auto", "equal", scalar or None, optional Aspect ratio of plot, so that ``aspect * size`` gives the width in inches. Only used if a ``size`` is provided. xincrease : bool or None, default: True Should the values on the x axes be increasing from left to right? if None, use the default for the matplotlib function. yincrease : bool or None, default: True Should the values on the y axes be increasing from top to bottom? if None, use the default for the matplotlib function. add_legend : bool or None, optional If True use xarray metadata to add a legend. add_colorbar : bool or None, optional If True add a colorbar. add_labels : bool or None, optional If True use xarray metadata to label axes add_title : bool or None, optional If True use xarray metadata to add a title subplot_kws : dict, optional Dictionary of keyword arguments for matplotlib subplots. Only applies to FacetGrid plotting. xscale : {'linear', 'symlog', 'log', 'logit'} or None, optional Specifies scaling for the x-axes. yscale : {'linear', 'symlog', 'log', 'logit'} or None, optional Specifies scaling for the y-axes. xticks : ArrayLike or None, optional Specify tick locations for x-axes. yticks : ArrayLike or None, optional Specify tick locations for y-axes. xlim : ArrayLike or None, optional Specify x-axes limits. ylim : ArrayLike or None, optional Specify y-axes limits. cmap : matplotlib colormap name or colormap, optional The mapping from data values to color space. Either a Matplotlib colormap name or object. If not provided, this will be either ``'viridis'`` (if the function infers a sequential dataset) or ``'RdBu_r'`` (if the function infers a diverging dataset). See :doc:`Choosing Colormaps in Matplotlib ` for more information. If *seaborn* is installed, ``cmap`` may also be a `seaborn color palette `_. Note: if ``cmap`` is a seaborn color palette, ``levels`` must also be specified. vmin : float or None, optional Lower value to anchor the colormap, otherwise it is inferred from the data and other keyword arguments. When a diverging dataset is inferred, setting `vmin` or `vmax` will fix the other by symmetry around ``center``. Setting both values prevents use of a diverging colormap. If discrete levels are provided as an explicit list, both of these values are ignored. vmax : float or None, optional Upper value to anchor the colormap, otherwise it is inferred from the data and other keyword arguments. When a diverging dataset is inferred, setting `vmin` or `vmax` will fix the other by symmetry around ``center``. Setting both values prevents use of a diverging colormap. If discrete levels are provided as an explicit list, both of these values are ignored. norm : matplotlib.colors.Normalize, optional If ``norm`` has ``vmin`` or ``vmax`` specified, the corresponding kwarg must be ``None``. extend : {'neither', 'both', 'min', 'max'}, optional How to draw arrows extending the colorbar beyond its limits. If not provided, ``extend`` is inferred from ``vmin``, ``vmax`` and the data limits. levels : int or array-like, optional Split the colormap (``cmap``) into discrete color intervals. If an integer is provided, "nice" levels are chosen based on the data range: this can imply that the final number of levels is not exactly the expected one. Setting ``vmin`` and/or ``vmax`` with ``levels=N`` is equivalent to setting ``levels=np.linspace(vmin, vmax, N)``. **kwargs : optional Additional arguments to wrapped matplotlib function Returns ------- artist : The same type of primitive artist that the wrapped matplotlib function returns   __module____name__ __qualname____doc__assignedNT)!r6r7r`r8 hue_style markersize linewidthrnrorprqrrMrrrr add_colorbar add_labels add_titlerrrrrrrrcmapvminvmaxnormextendlevelsr-rr4r0rsrtrrrbool | Iterable[bool]rrur1rstr | Colormap | NoneNormalize | Noner/)$r5rr6r7r`r8rrrrnrorprqrrMrrrrrrrrrrrrrrrrrrrrrr9c!: s|dkrt}|s| rr|dk r*|jddt}$|$|$d|$dtj|$d<t|fddi|$S|jdks|j dkrt d |"d kr|#d d }"|"r\d |#kst d }%|dkst |"d}t |"d kr|dkst |"d }t |"dkr|dkst |"d}t |"dkr4|dks,t |"d}t |"dkrLt |%ntj|%tdd~"|#dd}&jdkr|}'t}(n|}'t}(t||||'d})t||)j}*|*dd}+|*dd},|*dd}-|#j|-d|*dd}.|*dd}/t|.d}0|#j|0jdt|/|(|&d}1|#j|1jd|#d i}2|#d!i}3|0jdk r|0jsn|3j|0j|0jd"|#d#|0j}!ttd$|0jjft\}3|2sfd%d&d'D}4|2jf|4|dk r | dkr|jddt| | || f|} t}5t |5j!j"t d(kr| j#d)d)d*n| j#d)d)dd+nt| | || f|} |+|,f| |d,|2|#}6t$%t$&|rr|rr| '|(t)|0|1||jd-\}7}8|7rd.|3krt*|0j|3d.<t+|6| |#d/d|3|8rdjd0krt,|s|7s|0ntd|1|6| jd1nh|0jdk rt-td$|0j.}9n|0jg}9jd2krN| j/|6d3|9t*|0jd4n| j/|6|9t*|0jd4t0| |||||||| |6S)5N3d) projectionrvr5rrZplot1drrwrIrYUsing positional arguments is deprecated for plot methods, use keyword arguments instead.r:rx stacklevel _is_facetgridFscatter)r6r`r8rMr6r7r`)zpltr8rM)data)rE)rwidthr)sizepltcmap_params_subset cbar_kwargs)rticksrr-csi|]}||qSrIrI)rRvv cmap_paramsrIrJrUsz0_plot1d..newplotfunc..)rrrr3.5.0r)azimelev)rrZ vertical_axis)rqr)r_labelcbar_ax)rr|)Z legend_axrr~rar)1rirjrrrglobalsrrr@rMr{rr=r?warningswarnDeprecationWarningrrrmrrVrZdata_is_numericrrr]rrr rr r matplotlib __version__Z view_initrcanyasarrayrrrr!rrrrrr):r5r6r7r`r8rrrrnrorprqrrMrrrrrrrrrrrrrrrrrrrrrrrvrmsgrZsize_Zsize_rrNpltsrFrGrrErZ hueplt_normZ sizeplt_normrrZckwpltrZ add_colorbar_Z add_legend_Zhueplt_norm_valuesrrrJ newplotfuncUs(,                                 z_plot1d..newplotfuncr functoolswraps __wrapped__rZ commondocrrIrrJ_plot1dsVz^YrrzIterable[DataArray]zIterable[bool]r")rdarrayssuffixes rotate_labelsrqr9c Cst|tr|gdn|}td||||D]\}}}}} |dkr@q(|rnt||d} | dk rnt|d|d| | r(t|jtjr(t|d|dD]} | d| d qq(dS) Nrr6r7r`rset_rget_ ticklabelsrr) isinstancerr\r!getattrrcrdrerrr) rrrrrqaxisZ add_labelr5suffixZ rotate_labelrrrIrIrJ _add_labels5s$  r)!r6r7r`r8rrrrrMrrqrnrorprrrrrrrrrrrrrrrrrrrrr0rrr/r#)$r5rr6r7r`r8rrrrrMrrqrnrorprrrrrrrrrrrrrrrrrrrrr9c!$OsdSrPrI$r5r6r7r`r8rrrrrMrrqrnrorprrrrrrrrrrrrrrrrrrrrrrvrIrIrJrSs'r) r6r7r`r8rrrrrMrrqrnrprrrrrrrrrrrrrrrrrrrrc!$OsdSrPrIrrIrIrJr}s') r6r7r`r8rrrrrMrrqrorprrrrrrrrrrrrrrrrrrrrc!$OsdSrPrIrrIrIrJrs'zDataArray | None)rFrGrqrr9c  st}d|ksd|krtd|dd}|dd}|dd}|j|dd d |dk rr|j|d |dk r|j|d t|jjtd krdddg} n dddg} t |||dfdd| D} dd| D} |j dd| D|} t || dd|| S)zfScatter variables against each other. Wraps :py:func:`matplotlib:matplotlib.pyplot.scatter`. urTz&u, v are not allowed in scatter plots.rNrEr edgecolorsw)r )c)srr6r`r7rcsg|] }|qSrIrIrWZ plts_dictrIrJ szscatter..cSsg|]}|dk r|qSrPrIrRprIrIrJrscSsg|]}|qSrI)rrrrIrIrJrs)r;r;r;)TFF) r r?rrjrrr rrrirr) rFrGrqrrvrrrErZ axis_orderZ plts_or_nonerrrIrrJrs(      c&sd}jd|_tjddddddddddddddddddddddddddddddddddd d d d d d d d dd d dddddd d dd ddddddddddddddd d d#fdd}|`|S)z'Decorator for common 2d plotting logic.a Parameters ---------- darray : DataArray Must be two-dimensional, unless creating faceted plots. x : Hashable or None, optional Coordinate for *x* axis. If ``None``, use ``darray.dims[1]``. y : Hashable or None, optional Coordinate for *y* axis. If ``None``, use ``darray.dims[0]``. figsize : Iterable or float or None, optional A tuple (width, height) of the figure in inches. Mutually exclusive with ``size`` and ``ax``. size : scalar, optional If provided, create a new figure for the plot with the given size: *height* (in inches) of each plot. See also: ``aspect``. aspect : "auto", "equal", scalar or None, optional Aspect ratio of plot, so that ``aspect * size`` gives the *width* in inches. Only used if a ``size`` is provided. ax : matplotlib axes object, optional Axes on which to plot. By default, use the current axes. Mutually exclusive with ``size`` and ``figsize``. row : Hashable or None, optional If passed, make row faceted plots on this dimension name. col : Hashable or None, optional If passed, make column faceted plots on this dimension name. col_wrap : int, optional Use together with ``col`` to wrap faceted plots. xincrease : None, True, or False, optional Should the values on the *x* axis be increasing from left to right? If ``None``, use the default for the Matplotlib function. yincrease : None, True, or False, optional Should the values on the *y* axis be increasing from top to bottom? If ``None``, use the default for the Matplotlib function. add_colorbar : bool, optional Add colorbar to axes. add_labels : bool, optional Use xarray metadata to label axes. vmin : float or None, optional Lower value to anchor the colormap, otherwise it is inferred from the data and other keyword arguments. When a diverging dataset is inferred, setting `vmin` or `vmax` will fix the other by symmetry around ``center``. Setting both values prevents use of a diverging colormap. If discrete levels are provided as an explicit list, both of these values are ignored. vmax : float or None, optional Upper value to anchor the colormap, otherwise it is inferred from the data and other keyword arguments. When a diverging dataset is inferred, setting `vmin` or `vmax` will fix the other by symmetry around ``center``. Setting both values prevents use of a diverging colormap. If discrete levels are provided as an explicit list, both of these values are ignored. cmap : matplotlib colormap name or colormap, optional The mapping from data values to color space. If not provided, this will be either be ``'viridis'`` (if the function infers a sequential dataset) or ``'RdBu_r'`` (if the function infers a diverging dataset). See :doc:`Choosing Colormaps in Matplotlib ` for more information. If *seaborn* is installed, ``cmap`` may also be a `seaborn color palette `_. Note: if ``cmap`` is a seaborn color palette and the plot type is not ``'contour'`` or ``'contourf'``, ``levels`` must also be specified. center : float, optional The value at which to center the colormap. Passing this value implies use of a diverging colormap. Setting it to ``False`` prevents use of a diverging colormap. robust : bool, optional If ``True`` and ``vmin`` or ``vmax`` are absent, the colormap range is computed with 2nd and 98th percentiles instead of the extreme values. extend : {'neither', 'both', 'min', 'max'}, optional How to draw arrows extending the colorbar beyond its limits. If not provided, ``extend`` is inferred from ``vmin``, ``vmax`` and the data limits. levels : int or array-like, optional Split the colormap (``cmap``) into discrete color intervals. If an integer is provided, "nice" levels are chosen based on the data range: this can imply that the final number of levels is not exactly the expected one. Setting ``vmin`` and/or ``vmax`` with ``levels=N`` is equivalent to setting ``levels=np.linspace(vmin, vmax, N)``. infer_intervals : bool, optional Only applies to pcolormesh. If ``True``, the coordinate intervals are passed to pcolormesh. If ``False``, the original coordinates are used (this can be useful for certain map projections). The default is to always infer intervals, unless the mesh is irregular and plotted on a map projection. colors : str or array-like of color-like, optional A single color or a sequence of colors. If the plot type is not ``'contour'`` or ``'contourf'``, the ``levels`` argument is required. subplot_kws : dict, optional Dictionary of keyword arguments for Matplotlib subplots. Only used for 2D and faceted plots. (see :py:meth:`matplotlib:matplotlib.figure.Figure.add_subplot`). cbar_ax : matplotlib axes object, optional Axes in which to draw the colorbar. cbar_kwargs : dict, optional Dictionary of keyword arguments to pass to the colorbar (see :meth:`matplotlib:matplotlib.figure.Figure.colorbar`). xscale : {'linear', 'symlog', 'log', 'logit'} or None, optional Specifies scaling for the x-axes. yscale : {'linear', 'symlog', 'log', 'logit'} or None, optional Specifies scaling for the y-axes. xticks : ArrayLike or None, optional Specify tick locations for x-axes. yticks : ArrayLike or None, optional Specify tick locations for y-axes. xlim : ArrayLike or None, optional Specify x-axes limits. ylim : ArrayLike or None, optional Specify y-axes limits. norm : matplotlib.colors.Normalize, optional If ``norm`` has ``vmin`` or ``vmax`` specified, the corresponding kwarg must be ``None``. **kwargs : optional Additional keyword arguments to wrapped Matplotlib function. Returns ------- artist : The same type of primitive artist that the wrapped Matplotlib function returns. rrrNTF) r6r7rrMrrqrnrorprrrrrrrcenterrobustrrinfer_intervalscolorsrrrrrrrrrrrr-rr4rrrtrsrrrr/rstr | ArrayLike | Nonerur1r)#r5rr6r7rrMrrqrnrorprrrrrrrrrrrrrrrrrrrrrrrrvr9c < s6|!rbd}#|dkst|!d}t|!dkr<|dks4t|!d}t|!dkrRt|#ntj|#tdd~!| dkrd} jdksjdkr|dkrd } jd ko|jd |dk |dk k}$|$rd } |s|dk s|dk rt| |||}d \}}}|dkrt }jdkr<|" d d s<|dkr4ddl m }%~%d|d<d }&d }'|sH|rt}(|(d=|(d=|(|(dtj|(d<t|fddi|(S|jdks|jdkrtdt})jdkr|" d d s|dk rddl}*t||*jj std|"dd}+|+dk r(jd kr(tdn|+dk r@|$s@tdt||||$|+d\},}-||,}.||-}/|.jdks|/jdksjdkr|.|}.|/|}/|j}0n|/jd|.jdf}0|$r|-|,ftfdd|jD}0|0|jkr|j|0ddi}|.}1|/}2|j d d }3t!|1j\}4}5t!|2j\}6}7t"|4|6|3t#|3j$ftd |"d d i\}8}djkr|8d!|"d!<|8d"|"d"<t|t%rd|8d#<||"d$<d%jkr||"d&<||"d'<||"d(<d jkrt|t%rtd)t&||||f|}|4|6|3f||8d#|8d*|8d+|8d,d-|"}9| r|'t(||,|5|)t(||-|7|*|+jdkr|,t(|| r| rd.|krt(||d.<t-|9||||8}:n|dk s|rtd/d0|"krd} t.|| | |||||| t/0|4j1t/j2r2|3D]};|;4d1|;5d2q|9S)3Nrrr:rxrTcontoursurfaceFimshowr)NNFr)Axes3Drrr5 imshow_rgbrvrrZ dataarrayrwzEIf ax is passed to surface(), it must be created with projection="3d"rgbz,The "rgb" keyword is only valid for imshow()zVThe "rgb" keyword is only valid for imshow()with a three-dimensional array (per facet))r5r6r7rrc3s|]}|kr|VqdSrPrI)rRdZyx_dimsrIrJrYsz/_plot2d..newplotfunc..r<)rrrrrr}rrrz6plt.imshow's `aspect` kwarg is not available in xarrayrrr)rqrrrrrz>cbar_ax and cbar_kwargs can't be used with add_colorbar=False.originrr)6rr=r?rrrrr@rZas_numpyrir]Zmpl_toolkits.mplot3drrrrjrrrrMr{r mpl_toolkitsrZmplot3drZbroadcast_liker>r[rBrZto_masked_arrayrrrrstrrrr!rrrZ set_zlabelrrrcrdrerrrr)ZxvalnpZyvalnpZzvalrFZ xlab_extrarGZ ylab_extrarrZcbarrrrrJrs6*               $                  z_plot2d..newplotfuncrrrIrrJ_plot2dsTz\~r"F)rrMrrqrnrorprrrrrrrrrrrrrrrrrrrrrrrrrr))"r5r6r7rrMrrqrnrorprrrrrrrrrrrrrrrrrrrrrrrrvr9c"KsdSrPrI"r5r6r7rrMrrqrnrorprrrrrrrrrrrrrrrrrrrrrrrrrvrIrIrJrs&r)rrMrrqrnrprrrrrrrrrrrrrrrrrrrrrrrrc"KsdSrPrIr#rIrIrJrs&)rrMrrqrorprrrrrrrrrrrrrrrrrrrrrrrrc"KsdSrPrIr#rIrIrJrs&z np.ndarrayznp.ma.core.MaskedArray)r6r7r`rqrvr9cKs|jdks|jdkrtddd}||\}}||\}} ddd} t|dsXd | d <| || d dkr||| |g| d <n|||| g| d <|jd kr|jdd krtjj|jddd|jd} t |jtj r| d9} tjj || fdd}n| }d|tj |jdddf<|j|f| } d|fd|ffD]P\} }t |jtr6t|d| dtt|t|d| d|q6| S)az Image plot of 2D DataArray. Wraps :py:func:`matplotlib:matplotlib.pyplot.imshow`. While other plot methods require the DataArray to be strictly two-dimensional, ``imshow`` also accepts a 3D array where some dimension can be interpreted as RGB or RGBA color channels and allows this dimension to be specified via the kwarg ``rgb=``. Unlike :py:func:`matplotlib:matplotlib.pyplot.imshow`, which ignores ``vmin``/``vmax`` for RGB(A) data, xarray *will* use ``vmin`` and ``vmax`` for RGB(A) data by applying a single scaling factor and offset to all bands. Passing ``robust=True`` infers ``vmin`` and ``vmax`` :ref:`in the usual way `. Additionally the y-axis is not inverted by default, you can restore the matplotlib behavior by setting `yincrease=False`. .. note:: This function needs uniformly spaced coordinates to properly label the axes. Call :py:meth:`DataArray.plot` to check. The pixels are centered on the coordinates. For example, if the coordinate value is 3.2, then the pixels for those coordinates will be centered on 3.2. r:zBimshow requires 1D coordinates, try using pcolormesh or contour(f)cSsht|jtrdt|dfSzd|d|d}Wntk rNd}YnX|d||d|fS)z%Center the pixels on the coordinates.gg?r:rg?ra)rcrdrer!r= IndexError)r6ZxsteprIrIrJ_center_pixels%s zimshow.._center_pixelsupperZnearest)r interpolationrautorrZextentrraNrx)r:)re)rrr6r7rrr)r@r?hasattrrjshapercmaZonesrerdintegerZ concatenaterrmaskrr!rZaranger=)r6r7r`rqrvr%leftrtopZbottomdefaultsalpharrrTrIrIrJrs8         r(c"KsdSrPrIr#rIrIrJras&rc"KsdSrPrIr#rIrIrJrs&c"KsdSrPrIr#rIrIrJrs&cKs|j|||f|}|S)zc Contour plot of 2D DataArray. Wraps :py:func:`matplotlib:matplotlib.pyplot.contour`. )rr6r7r`rqrvrrIrIrJrs c"KsdSrPrIr#rIrIrJcontourfs&r4c"KsdSrPrIr#rIrIrJr4s&c"KsdSrPrIr#rIrIrJr4;s&cKs|j|||f|}|S)zk Filled contour plot of 2D DataArray. Wraps :py:func:`matplotlib:matplotlib.pyplot.contourf`. )r4r3rIrIrJr4ds r$c"KsdSrPrIr#rIrIrJr}qs&r}c"KsdSrPrIr#rIrIrJr}s&c"KsdSrPrIr#rIrIrJr}s&zScaleOptions | None)r6r7r`rqrrrvr9c Kst|}|dkr:t|dr6t|jdkr0d}q:d}nd}|rt|jtst|dt|dks|jdkrt|dt|dkrt|jdkrt |d|d}nt |d|d}t |d|d}|r8t|jts8t|dt|dkr8t|jdkrt |d|d}nt |d|d}t |d|d}| d|j |||f|}t|ds|jdkr|jdkr| |d|d | |d|d |S) zj Pseudocolor plot of 2D DataArray. Wraps :py:func:`matplotlib:matplotlib.pyplot.pcolormesh`. Nrr:TFr)Zcheck_monotonicscale)rr5ra)rcrr*r=r+rdrer!r@rZgridr}Zset_xlimZset_ylim) r6r7r`rqrrrrvrrIrIrJr}sJ     $r+c"KsdSrPrIr#rIrIrJr0 s&rc"KsdSrPrIr#rIrIrJrY s&c"KsdSrPrIr#rIrIrJr s&cKs|j|||f|}|S)zy Surface plot of 2D DataArray. Wraps :py:meth:`matplotlib:mpl_toolkits.mplot3d.axes3d.Axes3D.plot_surface`. )Z plot_surfacer3rIrIrJr s )rL)N)T)NN)NN)NN)NN)NN)NN)NN)NN)NN)NN)NN)NN)NNN)NN)NN)NN)Y __future__rrrtypingrrrrrrr r r ZnumpyrcZpandasrZpackaging.versionr Zxarray.core.alignmentr Zxarray.core.concatrZxarray.plot.facetgridrZxarray.plot.utilsrrrrrrrrrrrrrrrrr r!Zmatplotlib.axesr"Zmatplotlib.collectionsr#r$Zmatplotlib.colorsr%r&Zmatplotlib.containerr'Zmatplotlib.contourr(Zmatplotlib.imager)Zmpl_toolkits.mplot3d.art3dr*r+Z numpy.typingr,Zxarray.core.dataarrayr-Zxarray.core.typesr.r/r0r1r2r3rKr^rmrr|rr~rrrr"rrr4r}rrIrIrIrJs8 ,     P       X%*"[@>> >$ $ $"A0M`Z)Z)Z)1V(V(V(_V(V(V( V(V(V( V(V(V( CV(V(V(