U c@sdZddlmZddlmZddlmZddlZddlZ ddl m Z ddl m Z mZddlmZdd d Zd d ZddZdddddZdS)z/Common utility functions for rolling operations) annotations) defaultdict)castN)find_stack_level) ABCDataFrame ABCSeries) MultiIndexFcsttr,ttr,t\}}||Sttrddlmfdd}ittr|dkrRkrttjD].}j dd|fj dd|f|<q|Sjj st djj st dj dd \}}t||\}}j j}|D]0} | |kr | |kr || || | <q |j|d Sq|d krttttjD]x}ttjD]b} | |krkr| ||| <n2tj dd|fj dd| f|| <qqrdd lmj j} t| r<fd dttjDd ddj_jjdkrttj_t| tt| } fddtjjD} tjj| jg}tj| | f|d_tjj}|dgtt|d}|nHtttjtt| g_ddt| gjg_n$tjjggggdjddj jj_j | jjj_Sn*fddttjD|Sn t!SdS)Nr DataFramecs.||jd}t|jdkr*|j|j|_|S)N)indexr)r lencolumns)dataZframe_templateresultr =/tmp/pip-unpacked-wheel-g7fro6k3/pandas/core/window/common.pydataframe_from_int_dictsz3flex_binary_moment..dataframe_from_int_dictFz'arg1' columns are not uniquez'arg2' columns are not uniqueouter)join)r r T)concatcs2g|]*fddttjDddqS)csg|]}|qSrr).0j)iresultsrr Lsz1flex_binary_moment...T) ignore_index)ranger r )r)arg2rr)rrrJs z&flex_binary_moment..)rZaxisc3s0|](}tj|ttjVqdSN)nprepeatr Zget_level_valuesr rr)rrrr cs  z%flex_binary_moment..)names)levelscodesZfloat64)r r Zdtypecs*i|]"}|tjdd|fqSr) prep_binaryilocr")arg1rfrr sz&flex_binary_moment..)" isinstancerr'rZpandasr rr r r(Z is_unique ValueErrorZalignunionr rdictrZnlevelsrrr Ztilelistr$nameZ from_arraysr%Zreorder_levelsZ sort_indexZ from_productZ swaplevelZ set_namesflex_binary_moment)r)rr*ZpairwiseXYrrZ res_columnscolrZ result_indexZ result_levelZ arg2_levelsZ result_namesZ num_levelsZ new_orderr)r r)rrr*rrrr2s      ,  $             r2c Cs\tjddt|}|dk}W5QRXt|trH|jrXd||<n|rXd||<|S)Nignore)allr)r Zerrstatesqrtr,rZ_valuesany)xrmaskrrrzsqrts    r<cCs |d|}|d|}||fS)Nrr)r)rr3r4rrrr's  r'strNone)kernelreturncCs|dk ot|dk}|dk o&t|dk}|r6|r6d}n|r@d}n|rJd}nd}|dkr~tjd|d|jd |d ttd dS) a Warn for deprecation of args and kwargs in rolling/expanding functions. Parameters ---------- cls : type Class to warn about. kernel : str Operation name. args : tuple or None args passed by user. Will be None if and only if kernel does not have args. kwargs : dict or None kwargs passed by user. Will be None if and only if kernel does not have kwargs. Nrzargs and kwargsargskwargszPassing additional z to .zj has no impact on the result and is deprecated. This will raise a TypeError in a future version of pandas.)category stacklevel)r warningswarn__name__ FutureWarningr)clsr?rArBZ warn_argsZ warn_kwargsmsgrrrmaybe_warn_args_and_kwargssrM)F)__doc__ __future__r collectionsrtypingrrGZnumpyr Zpandas.util._exceptionsrZpandas.core.dtypes.genericrrZpandas.core.indexes.apirr2r<r'rMrrrrs