4ik| UdZddlmZddlZddlZddlZddlZddlZddlZddl m Z m Z m Z m Z ddlmZddlmZddlmZddlmZmZmZmZmZmZddlZddlmcmZdd lm Z m!Z!dd l"m#Z#dd l$m%Z%m&Z&m'Z'ejPd k\r dd lm)Z)m*Z*m+Z+n dd l,m)Z)m*Z*m+Z+ejPdk\r ddlm-Z-m.Z.nddl,m-Z-m.Z.er ddl/Z0ddl"m1Z1ddl2m3Z4ddl5m6Z7eddZ8ede#Z9edZ:e.dZ;edZ<e*de dee*dde:eZY dUd?ZZdVd@Z[e dW dXdAZ\e dYdBZ\ dZ d[dCZ\d\d]dDZ]edEZ^dFeQdG< GdHdFZ_dIZ` d^dJZad_dKZbycc}}w)`zUtility routines.) annotationsN)CallableIteratorMappingMutableMapping)deepcopy)groupby) itemgetter) TYPE_CHECKINGAnyLiteralTypeVarcastoverload)is_pandas_dataframeis_polars_dataframe) IntoDataFrame) SchemaBase SchemaLike Undefined) )Protocol TypeAliasTyperuntime_checkable)r ) Concatenate ParamSpec)IntoExpr) DataFrame)StandardType_T_PandasDataFrameTz pd.DataFrame)boundTIntoDataFrameTPR WrapsFunc.) type_params WrappedFunc WrapsMethodz Callable[Concatenate[T, ...], R] WrappedMethodc"eZdZ d ddZy) DataFrameLikecyN)self nan_as_null allow_copys [/home/obispo/Crisostomo_bridge/mision_env/lib/python3.12/site-packages/altair/utils/core.py __dataframe__zDataFrameLike.__dataframe__>sN)FT)r3boolr4r8return DfiDataFrame)__name__ __module__ __qualname__r6r1r7r5r.r.<s%=A59 r7r.ONQG)ordinalnominal quantitativetemporalgeojson)argmaxargminaveragecountdistinctmaxmeanmedianminmissingproductq1q3ci0ci1stderrstdevstdevpsumvalidvaluesvariance variancep exponential exponentialb) row_numberrank dense_rank percent_rank cume_distntilelaglead first_value last_value nth_value)Qyearquartermonthweekday dayofyeardatehoursminutesseconds milliseconds yearquarteryearquartermonth yearmonth yearmonthdateyearmonthdatehoursyearmonthdatehoursminutes yearmonthdatehoursminutessecondsyearweek yearweekdayyearweekdayhoursyearweekdayhoursminutesyearweekdayhoursminutesseconds yeardayofyear quartermonth monthdatemonthdatehoursmonthdatehoursminutesmonthdatehoursminutessecondsweekday weeksdayhours weekdayhoursweekdayhoursminutesweekdayhoursminutessecondsdayhoursdayhoursminutesdayhoursminutesseconds hoursminuteshoursminutessecondsminutessecondssecondsmillisecondsutcyear utcquarterutcmonthutcweekutcday utcdayofyearutcdateutchours utcminutes utcsecondsutcmillisecondsutcyearquarterutcyearquartermonth utcyearmonthutcyearmonthdateutcyearmonthdatehoursutcyearmonthdatehoursminutes#utcyearmonthdatehoursminutesseconds utcyearweekutcyearweekdayutcyearweekdayhoursutcyearweekdayhoursminutes!utcyearweekdayhoursminutessecondsutcyeardayofyearutcquartermonth utcmonthdateutcmonthdatehoursutcmonthdatehoursminutesutcmonthdatehoursminutesseconds utcweekdayutcweekdayhoursutcweekdayhoursminutesutcweekdayhoursminutesseconds utcdayhoursutcdayhoursminutesutcdayhoursminutessecondsutchoursminutesutchoursminutessecondsutcminutessecondsutcsecondsmillisecondsz (?P.*)z (?P{})|z(?Pcount)z (?Pcount)z(?P{})z (?P{})z(?P{}))fieldtype agg_countop_count aggregate window_optimeUnit)rrrrz mixed-integermixed-integer-floatcomplexintegerfloatingrD categoricalcatrB>bytesmixedstringbooleanunicoderrC>rqtimeperioddatetime timedelta datetime64 timedelta64rEz.I don't know how to infer vegalite type from 'z'. Defaulting to nominal. stacklevel) pandas.api.typesrhasattrrordered categoriestolistwarningswarn)datartyps r5infer_vegalite_type_for_pandasrs- d5 )C  '$"6488;K;K488..55788 Q Q   .to_list_if_arrayxs" c2:: &::< Jr7rrcategoryrr8r)r timestampc"|jSr0) isoformat)xs r5z+sanitize_pandas_dataframe..s Q[[]r7NaTrField " " has type "^" which is not supported by Altair. Please convert to either a timestamp or a numerical value.r> Int8Int16Int32Int64UInt8UInt16UInt32UInt64Float32Float64)rpandasrcopyrcolumns RangeIndexastyper ValueErrorindex MultiIndexdtypesitemsobjectwherenotnullrapplyreplacer rrisnullisinf) dfpdcol_namemsgr dtype_itemr  dtype_namecol bad_valuesr s @r5sanitize_pandas_dataframerBMs_( !2779 -B"**bmm,ZZ&&s+ JJ"(C(:8,G// S/ ! ""((BMM*2o"**bmm,2o iioo'O: z!}-1 Z  #X,%%f-C99S[[]D9BxL 8 #X,%%f-C99S[[]D9BxL 6 !h<..v6BxL 9 $X,%%f-C99S[[]D9BxL  " "#< =8 ""#:;CCE2N xL " "; /(<w7  S/ !  " ": .    X,%%f-C99S[[]D9BxL eRZZ 0h<..v6BxL eR[[ 1X,C 5J::f-33ZKFBxL f_X,%%f-334DEC99S[[]D9BxL_O:` Ir7cF|j}g}d}t|j}|jD]T\}}|tj k(r]|r[|j t j|jtjjj|w|tj k(r>|j t j|jj||tjk(rB|j t j|jj|d|tjk(rd|d|d}t||j |W|j|S)z3Sanitize narwhals.DataFrame for JSON serialization.z%Y-%m-%dT%H:%M:%Sz%.frrr)schemar to_nativer2nwDateappendr@rDatetimedt to_stringDurationr.select)rrDr+local_iso_fmt_string is_polarsnamer r=s r5sanitize_narwhals_dataframerQsE[[F G.#DNN$45I||~! e BGG   NNt !!"++.11;;>> import pandas as pd >>> data = pd.DataFrame({"foo": ["A", "B", "A", "B"], "bar": [1, 2, 3, 4]}) >>> parse_shorthand("name") == {"field": "name"} True >>> parse_shorthand("name:Q") == {"field": "name", "type": "quantitative"} True >>> parse_shorthand("average(col)") == {"aggregate": "average", "field": "col"} True >>> parse_shorthand("foo:O") == {"field": "foo", "type": "ordinal"} True >>> parse_shorthand("min(foo):Q") == { ... "aggregate": "min", ... "field": "foo", ... "type": "quantitative", ... } True >>> parse_shorthand("month(col)") == { ... "field": "col", ... "timeUnit": "month", ... "type": "temporal", ... } True >>> parse_shorthand("year(col):O") == { ... "field": "col", ... "timeUnit": "year", ... "type": "ordinal", ... } True >>> parse_shorthand("foo", data) == {"field": "foo", "type": "nominal"} True >>> parse_shorthand("bar", data) == {"field": "bar", "type": "quantitative"} True >>> parse_shorthand("bar:O", data) == {"field": "bar", "type": "ordinal"} True >>> parse_shorthand("sum(bar)", data) == { ... "aggregate": "sum", ... "field": "bar", ... "type": "quantitative", ... } True >>> parse_shorthand("count()", data) == { ... "aggregate": "count", ... "type": "quantitative", ... } True r) is_data_typez{agg_count}\(\)z{aggregate}\({field}\)z{op_count}\(\)z{window_op}\({field}\)z{timeUnit}\({field}\)z{field}c3*K|] }|dz|f yw)z:{type}Nr1.0ps r5 z"parse_shorthand..~s)O1y=!*<)Osc 3K|]B}tjd|jditzdztjDyw)z\Az\ZNr1)recompileformatSHORTHAND_UNITSDOTALLras r5rdz"parse_shorthand..s?  581886o66> JsAA c3~K|]4}|j!|jj6ywr0)match groupdict)rbexp shorthands r5rdz"parse_shorthand..s: yy#/ IIi * * , s:=rrrJrDrrEr\rTrSrsort:z"{}" z0is not one of the valid encoding data types: {}.z, zi For more details, see https://altair-viz.github.io/user_guide/encodings/index.html#encoding-data-types. z>If you are trying to use a column name that contains a colon, zPprefix it with a backslash; for example "column\:name" instead of "column:name".)rZr_extendlist itertoolschainrdictnextINV_TYPECODE_MAPgetr7rFrXrDObjectUnknownrrEr infer_vegalite_type_for_narwhalstuplerfindr.rhsplitjoin TYPECODE_MAPr[)rorparse_aggregatesparse_window_opsparse_timeunits parse_typesr_patternsregexpsattrsunescaped_fieldr[rDcolumns` r5parse_shorthandrsL/  H+,-234*+,234123 OOZL! )Oh)OPQG )T"   (,,U6]E&MJf  g&&&f UvU2"f U|D1.00r:..F f $_-Fo&  +&g&7&7&9: >v?O?O?Q Rf @ Hf %-/ %f a 0f %f a 0f 5 5> ! 'N5>//4q8 9T A NN5>//4R8 9@GG ,--/0 {  { O  O b  b   Lr7c|j}tj|r<|jj x}j sd|j fS|tjk(s&|tjk(s|tjk(ry|jry|tjk(s|tjk(ryd|}t|)NrBrCrDrEzUnexpected DtypeKind: )r rFis_ordered_categoricalrget_categoriesis_emptyto_listString CategoricalBoolean is_numericrIrGr.)rr rr=s r5r~r~s LLE !!&)%zz88::DDF*,,...  Ubnn48K     "++ "''!1&ug.or7cHtdd}tdd}dfd }|S)a_ Use the signature and doc of ``tp`` for the decorated callable ``cb``. - **Overload 1**: Decorating method - **Overload 2**: Decorating function Returns ------- **Adding the annotation breaks typing**: Overload[Callable[[WrapsMethod[T, R]], WrappedMethod[T, P, R]], Callable[[WrapsFunc[R]], WrappedFunc[P, R]]] cyr0r1cbs r5decoratezuse_signature..decoratesFIr7cyr0r1rs r5rzuse_signature..decorates.decorates!Z4ZZ 6  H(9"++a&HILF&!P6+<+>> original = {"x": {"b": 2, "c": 4}} >>> update = {"x": {"b": 5, "d": 6}, "y": 40} >>> update_nested(original, update) # doctest: +SKIP {'x': {'b': 5, 'c': 4, 'd': 6}, 'y': 40} >>> original # doctest: +SKIP {'x': {'b': 5, 'c': 4, 'd': 6}, 'y': 40} )rr2rrr{rr)rrr*rrorig_vals r5rr sw> H%LLN S c7 #||C,H(N3 -h < # HSM  Or7ctj}|rddlm}|}nd}||j |yt j |y)Nr) get_ipython)sysexc_infoIPython.core.getipythonr showtraceback tracebackprint_exception) in_ipythonrrips r5display_tracebackr:s@||~H7 ]  ~ "!!8,r7)rdatumvalue _ChannelCache_CHANNEL_CACHEcLeZdZUded<ded<ed dZd dZd dZd dZy )rdict[type[SchemaBase], str]channel_to_name/dict[str, dict[_ChannelType, type[SchemaBase]]]name_to_channelc t}tS#t$rD|j|}t|_t |j|_|aYtSwxYwr0)r NameError__new___init_channel_to_namer_invert_group_channelsr)clscacheds r5 from_cachez_ChannelCache.from_cacheVs\ $#F   $[[%F%:%`] | type[``] | type[``], str] r)channels) altair.vegalite.v6.schemarFieldChannelMixinValueChannelMixinDatumChannelMixin__dict__r[rr issubclassr_encoding_name)chmixinscs r5rrs{9  ! !2#7#79M9M MF##%  a :a#8Z:=V 1    sAB cdd}t|jtd}|Dcic]\}}|||c}}Scc}}w)z;Grouped inverted index for `_ChannelCache.channel_to_name`.ci}|D]>\}}|j}|jdr$|jdrd}nd}|||<@|S)z Returns a 1-2 item dict, per channel. Never includes `datum`, as it is never utilized in `wrap_in_channel`. DatumValuerr)r;endswith)ititemr_rPsub_keys r5_reducez'_invert_group_channels.._reducesZ -/ EB;;D}}W%w'!!DM  r7r)rzIterator[tuple[type[Any], str]]r9r )r r2r )mrgrouperrchanss r5rrs@ $aggiA/G.5 6(!UAwu~  66 6sActj}|D]c}t|ttfrt t |dn|}|jt|}||vr|||<Td|d}t||j|S)aa Infer typed keyword arguments for args and kwargs. Parameters ---------- args : Sequence Sequence of function args kwargs : MutableMapping Dict of function kwargs Returns ------- kwargs : dict All args and kwargs in a single dict, with keys and types based on the channels mapping. Nz encoding z specified twice.) rrrrurryiterrrr.r)argsrcacheargrrr=s r5rrs"  $ $ &E"&0tUm&DT$s)T "#%%d2h/ 6 !"F8 h\):;CS/ !"  % %f --r7)rr r9z=InferredVegaLiteType | tuple[InferredVegaLiteType, list[Any]])rrr9r)rMutableMapping[Any, Any]r9r)r r r r r9r8)r:r"r9r")rnw.DataFrame[TIntoDataFrame]r9r)rrr9znw.DataFrame[Any])NTFTT)rozdict[str, Any] | strrzIntoDataFrame | Nonerr8rr8rr8rr8r9r)rz nw.Seriesr9z8InferredVegaLiteType | tuple[InferredVegaLiteType, list])rzCallable[P, Any]).)rrrMapping[Any, Any]r*zLiteral[False]r9r)rrrrr*z Literal[True]r9r)F)rr rrr*r8r9r)T)rr8)rrr9r)rztuple[Any, ...]rr)cr __future__rrvrrfrrrcollections.abcrrrrr*rr operatorr typingr r r rrrrnarwhals.stable.v1stablev1rFnarwhals.stable.v1.dependenciesrrnarwhals.stable.v1.typingraltair.utils.schemapirrr version_inforrrtyping_extensionsrrr)r;raltair.utils._dfi_typesr r:!altair.vegalite.v6.schema._typingr!InferredVegaLiteTyper"r$r%r&r'r(r*r+r,r.rr2rz AGGREGATESWINDOW_AGGREGATES TIMEUNITSrurwrVALID_TYPECODESrhrri frozensetrrrrrr rBrQr]rr~rrr _ChannelTyperrrr)rvs00r5rs " GGGGT3CCwAALLw--82AX 3>J)? CL cN CL +xQ'7aT J M8AqD>1vN 5Aq6 Xk!Q$/23!Q H  &2%7%7%9:TQAqD: : R  hytL'94@P;QRS  ! !#((?"; <'$++CHHZ,@A$$SXXj;L.L%MN"))#((9*=>89L - -B-`$BAH & ! B*"&!"  x#x xx x  x  xxv =.'T #&# # # # #  ## # # # #** * * *Z - 01  1 1 h07"7476.];s0 K%