2idZddlmZddlZddlZddlZddlZddl m Z m Z m Z ejeZGdde j ZGdde j$ZGd d e j(ZGd d Zee j.e<ee j.ej2<ee j.e<ee j.ej6<y) a Plotting of string "category" data: ``plot(['d', 'f', 'a'], [1, 2, 3])`` will plot three points with x-axis values of 'd', 'f', 'a'. See :doc:`/gallery/lines_bars_and_markers/categorical_variables` for an example. The module uses Matplotlib's `matplotlib.units` mechanism to convert from strings to integers and provides a tick locator, a tick formatter, and the `.UnitData` class that creates and stores the string-to-integer mapping. ) OrderedDictN)_apitickerunitscLeZdZedZedZedZedZy)StrCategoryConverterc0| tdtj|tjtj |t }|j|tj|jjtg|S)a Convert strings in *value* to floats using mapping information stored in the *unit* object. Parameters ---------- value : str or iterable Value or list of values to be converted. unit : `.UnitData` An object mapping strings to integers. axis : `~matplotlib.axis.Axis` The axis on which the converted value is plotted. .. note:: *axis* is unused. Returns ------- float or `~numpy.ndarray` of float zMissing category information for StrCategoryConverter; this might be caused by unintendedly mixing categorical and numeric datadtype)otypes) ValueErrorr_validate_unitnp atleast_1darrayobjectupdate vectorize_mapping __getitem__float)valueunitaxisvaluess J/var/www/html/acx/venv/lib/python3.12/site-packages/matplotlib/category.pyconvertzStrCategoryConverter.convertst* <   ++D1rxxV<= FFr||DMM55ugFvNNctj|t|j}t |j}t j ||S)a Set the default axis ticks and labels. Parameters ---------- unit : `.UnitData` object string unit information for value axis : `~matplotlib.axis.Axis` axis for which information is being set .. note:: *axis* is not used Returns ------- `~matplotlib.units.AxisInfo` Information to support default tick labeling )majlocmajfmt)rrStrCategoryLocatorrStrCategoryFormatterrAxisInfo)rrr r!s raxisinfozStrCategoryConverter.axisinfo<sA( ++D1$DMM2%dmm4~~VF;;rc|j&|jt||jS|jj||jS)aO Set and update the `~matplotlib.axis.Axis` units. Parameters ---------- data : str or iterable of str axis : `~matplotlib.axis.Axis` axis on which the data is plotted Returns ------- `.UnitData` object storing string to integer mapping )r set_unitsUnitDatar)datars r default_unitsz"StrCategoryConverter.default_unitsWsH" ::  NN8D> *zz JJ  d #zzrc:t|dstd|dy)NrzProvided unit "zU" is not valid for a categorical converter, as it does not have a _mapping attribute.)hasattrr )rs rrz#StrCategoryConverter._validate_unitns3tZ(!$(GGH H)rN)__name__ __module__ __qualname__ staticmethodrr%r*rrrrrsUOO@<<4,HHrrc"eZdZdZdZdZdZy)r"z1Tick at every integer mapping of the string data.c||_yz Parameters ---------- units_mapping : dict Mapping of category names (str) to indices (int). N_unitsself units_mappings r__init__zStrCategoryLocator.__init__x $ rcHt|jjSN)listr6r)r8s r__call__zStrCategoryLocator.__call__sDKK&&())rc|Sr=r1)r8vminvmaxs r tick_valueszStrCategoryLocator.tick_valuess v rN)r-r.r/__doc__r:r?rCr1rrr"r"vs;$*rr"c4eZdZdZdZddZdZedZy)r#z0String representation of the data at every tick.c||_yr4r5r7s rr:zStrCategoryFormatter.__init__r;rNc,|j|gdS)Nr) format_ticks)r8xposs rr?zStrCategoryFormatter.__call__s  !%a((rc|jjDcic]\}}||j|}}}|Dcgc]}|jt |dc}Scc}}wcc}w)N)r6items_textgetround)r8rkv r_mappingvals rrHz!StrCategoryFormatter.format_tickss\26++2C2C2EF$!QQ 1 %F F9?@# eCj"-@@G@s A&"A,ct|tr|jd}|St|ts t|}|S)z0Convert text values into utf-8 or ascii strings.zutf-8)encoding) isinstancebytesdecodestrrs rrNzStrCategoryFormatter._texts? eU #LL'L2E E3'JE rr=) r-r.r/rDr:r?rHr0rNr1rrr#r#s):$)A rr#c*eZdZddZedZdZy)r(Nc|t|_tj|_||j |yy)z Create mapping between unique categorical values and integer ids. Parameters ---------- data : iterable sequence of string values N)rr itertoolscount_counterr)r8r)s rr:zUnitData.__init__s2$  !)   KK  rc t|y#t$r9 tjj |Yy#tt f$rYYywxYwwxYw)zY Helper method to check whether a string can be parsed as float or date. FT)rr dateutilparserparse TypeError)rTs r_str_is_convertiblezUnitData._str_is_convertiblesY   #J   %%c* *   s$  A9A A A  Actjtj|t}d}t j |D]g}t jttf||r|j|}||jvsFt|j|j|<i|jr|rtj!dyyy)z Map new values to integer identifiers. Parameters ---------- data : iterable of str or bytes Raises ------ TypeError If elements in *data* are neither str nor bytes. r Tr[zUsing categorical units to plot a list of strings that are all parsable as floats or dates. If these strings should be plotted as numbers, cast to the appropriate data type before plotting.N)rrrrrfromkeysrcheck_isinstancerZrXrfrnextr`size_loginfo)r8r) convertiblerTs rrzUnitData.updates}}RXXd&9: ''- 9C  ! !3,c :"66s; $--'%)$--%8 c" 9 99 II? @%9rr=)r-r.r/r:r0rfrr1rrr(r(s!   @rr()rD collectionsrdateutil.parserrbr^loggingnumpyr matplotlibrrr getLoggerr-rlConversionInterfacerLocatorr" Formatterr#r(registryrZstr_rXbytes_r1rrr{s $**w"YH544YHx(6++<:@:@|+,s.0rww,.u02ryyr