U Gv fd@sddlmZddlmZddlmZmZddlZddlm Z ddl m Z ddl m Z ddlmZdd lmZmZdd lmZeGd d d eZeGd ddeZeGdddeZdS)) annotations) dataclass)ClassVarCallableN) DataFrame)Scale)GroupBy)Stat)EstimateAggregatorWeightedAggregator)Vectorc@s@eZdZUdZdZded<dZded<dd d d dd d dZdS)Agga` Aggregate data along the value axis using given method. Parameters ---------- func : str or callable Name of a :class:`pandas.Series` method or a vector -> scalar function. See Also -------- objects.Est : Aggregation with error bars. Examples -------- .. include:: ../docstrings/objects.Agg.rst meanstr | Callable[[Vector], float]funcTClassVar[bool]group_by_orientrrstrdict[str, Scale]datagroupbyorientscalesreturncCs8ddd|}||||jij|gdjdd}|S)NyxrrZsubsetTZdrop)getZaggrdropna reset_index)selfrrrrvarresr&>/tmp/pip-unpacked-wheel-ir0tye89/seaborn/_stats/aggregation.py__call__)sz Agg.__call__N)__name__ __module__ __qualname____doc__r__annotations__rr(r&r&r&r'r s   r c@sxeZdZUdZdZded<dZded<dZd ed <d Zd ed <dZ ded<dddddddZ ddddddddZ d S)Esta, Calculate a point estimate and error bar interval. For more information about the various `errorbar` choices, see the :doc:`errorbar tutorial `. Additional variables: - **weight**: When passed to a layer that uses this stat, a weighted estimate will be computed. Note that use of weights currently limits the choice of function and error bar method to `"mean"` and `"ci"`, respectively. Parameters ---------- func : str or callable Name of a :class:`numpy.ndarray` method or a vector -> scalar function. errorbar : str, (str, float) tuple, or callable Name of errorbar method (one of "ci", "pi", "se" or "sd"), or a tuple with a method name ane a level parameter, or a function that maps from a vector to a (min, max) interval. n_boot : int Number of bootstrap samples to draw for "ci" errorbars. seed : int Seed for the PRNG used to draw bootstrap samples. Examples -------- .. include:: ../docstrings/objects.Est.rst rrr)ci_zstr | tuple[str, float]errorbariintn_bootNz int | NoneseedTrrrrr )rr$ estimatorrcCs|||}t|gSN)pdr)r#rr$r5r%r&r&r'_process^s z Est._processrrrc Cs|j|jd}d|kr*t|j|jf|}nt|j|jf|}ddd|}|||j||j|gdj dd}| |d |||d ||i}|S) N)r3r4ZweightrrrrTrminmax) r3r4r rr1r applyr8r!r"Zfillna) r#rrrrZboot_kwsZenginer$r%r&r&r'r(fs$&z Est.__call__) r)r*r+r,rr-r1r3r4rr8r(r&r&r&r'r.7s      r.c@seZdZddZdS)RollingcCsdSr6r&)r#rrrrr&r&r'r(szRolling.__call__N)r)r*r+r(r&r&r&r'r<}sr<) __future__rZ dataclassesrtypingrrZpandasr7rZseaborn._core.scalesrZseaborn._core.groupbyrZseaborn._stats.baser Zseaborn._statisticsr r Zseaborn._core.typingr r r.r<r&r&r&r's       $E