o i@sdZddlmZddlZddlZddlZddlZddl m Z m Z m Z e eZGddde jZGddde jZGd d d e jZGd d d Zee je<ee jej<ee je<ee jej<dS) 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)cbooktickerunitsc@s0eZdZeddZeddZeddZdS)StrCategoryConvertercCsf|durtdttj|td}tdd|Dr"tj|tdS||tj |j j tgd|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 ndarray[float] NzMissing category information for StrCategoryConverter; this might be caused by unintendedly mixing categorical and numeric datadtypecss,|]}tj|ot|ttf VqdSN)rConversionInterface is_numlike isinstancestrbytes).0vrN/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/matplotlib/category.py 8s   z/StrCategoryConverter.convert..)otypes) ValueErrornp atleast_1darrayobjectallasarrayfloatupdate vectorize_mapping __getitem__)valueunitaxisvaluesrrrconverts zStrCategoryConverter.convertcCs"t|j}t|j}tj||dS)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 Returns ------- `~matplotlib.units.AxisInfo` Information to support default tick labeling .. note: axis is not used )majlocmajfmt)StrCategoryLocatorrStrCategoryFormatterrAxisInfo)r"r#r&r'rrraxisinfo@s  zStrCategoryConverter.axisinfocCs0|jdur|t||jS|j||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 N)r set_unitsUnitDatar)datar#rrr default_unitsYs  z"StrCategoryConverter.default_unitsN)__name__ __module__ __qualname__ staticmethodr%r+r/rrrrrs $ rc@s(eZdZdZddZddZddZdS) r(z1Tick at every integer mapping of the string data.cC ||_dS)z Parameters ----------- units_mapping : dict Mapping of category names (str) to indices (int). N_unitsself units_mappingrrr__init__s zStrCategoryLocator.__init__cCst|jSr )listr6r$r8rrr__call__|szStrCategoryLocator.__call__cCs|Sr r)r8vminvmaxrrr tick_valuesszStrCategoryLocator.tick_valuesN)r0r1r2__doc__r:r>rArrrrr(qs  r(c@s6eZdZdZddZd ddZddZed d ZdS) r)z0String representation of the data at every tick.cCr4)z Parameters ---------- units_mapping : dict Mapping of category names (str) to indices (int). Nr5r7rrrr:r;zStrCategoryFormatter.__init__NcCs||gdS)Nr) format_ticks)r8xposrrrr>szStrCategoryFormatter.__call__cs*fddjDfdd|DS)Ncsi|] \}}||qSr)_text)rkrr=rr z5StrCategoryFormatter.format_ticks..csg|] }t|dqS))getround)rval) r_mappingrr rIz5StrCategoryFormatter.format_ticks..)r6items)r8r$r)rNr8rrCsz!StrCategoryFormatter.format_tickscCs0t|tr |jdd}|St|tst|}|S)z0Convert text values into utf-8 or ascii strings.zutf-8)encoding)r rdecoder r!rrrrFs   zStrCategoryFormatter._textr ) r0r1r2rBr:r>rCr3rFrrrrr)s r)c@s*eZdZdddZeddZddZdS) r-NcCs,t|_t|_|dur||dSdS)z Create mapping between unique categorical values and integer ids. Parameters ---------- data : iterable sequence of string values N)rr itertoolscount_counterr)r8r.rrrr:s  zUnitData.__init__c CsNzt|WdSty&z tj|WYdSttfy%YYdSww)zY Helper method to check whether a string can be parsed as float or date. FT)rrdateutilparserparse TypeError)rMrrr_str_is_convertibles  zUnitData._str_is_convertiblecCsxttj|td}d}t|D]}tjtt f|d|r#| |}||j vr0t |j |j |<q|r:tddSdS)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. rTrSzUsing 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)rrrrrfromkeysr_check_isinstancer rr[rnextrV_loginfo)r8r. convertiblerMrrrrs   zUnitData.updater )r0r1r2r:r3r[rrrrrr-s    r-)rB collectionsrdateutil.parserrWrTloggingnumpyr matplotlibrrr getLoggerr0r_r rLocatorr( Formatterr)r-registryr str_rbytes_rrrrs   W >