jndZddlmZddlZddlZddlZddlZddl m Z m Z m Z m Z ejeZGdde jZGdde jZGd d e jZGd d Zexe je<xe jej<xe je<e 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)_apicbooktickerunitscfeZdZedZedZedZedZdS)StrCategoryConvertercv|tdt|tjtj|t }||tj|j j tg|}tj |s|n|dS)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 NzMissing category information for StrCategoryConverter; this might be caused by unintendedly mixing categorical and numeric datadtype)otypesr) ValueErrorr _validate_unitnp atleast_1darrayobjectupdate vectorize_mapping __getitem__floatris_scalar_or_string)valueunitaxisvaluesss AC:\Projects\DSCons\.venv\Lib\site-packages\matplotlib\category.pyconvertzStrCategoryConverter.converts* <    ++D111rxV<<<== F CBL2E7 C C CF K K1%88BqqadBct|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)r rStrCategoryLocatorrStrCategoryFormatterrAxisInfo)rrr#r$s raxisinfozStrCategoryConverter.axisinfo=sM( ++D111$DM22%dm44~VF;;;;r!c|j#|t|n|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 )r set_unitsUnitDatar)datars r default_unitsz"StrCategoryConverter.default_unitsXsE" :  NN8D>> * * * * J  d # # #zr!cLt|dstd|ddS)NrzProvided unit "zU" is not valid for a categorical converter, as it does not have a _mapping attribute.)hasattrr)rs rrz#StrCategoryConverter._validate_unitosPtZ(( HG$GGGHH H H Hr!N)__name__ __module__ __qualname__ staticmethodr r(r-rr!rr r sCC\CB<<\<4\,HH\HHHr!r c$eZdZdZdZdZdZdS)r%z1Tick at every integer mapping of the string data.c||_dSz Parameters ---------- units_mapping : dict Mapping of category names (str) to indices (int). N_unitsself units_mappings r__init__zStrCategoryLocator.__init__y$ r!cNt|jSN)listr9r)r;s r__call__zStrCategoryLocator.__call__sDK&&(()))r!c|Sr@r4)r;vminvmaxs r tick_valueszStrCategoryLocator.tick_valuess tvv r!N)r0r1r2__doc__r=rBrFr4r!rr%r%wsG;;$$$***r!r%c<eZdZdZdZddZdZedZdS)r&z0String representation of the data at every tick.c||_dSr7r8r:s rr=zStrCategoryFormatter.__init__r>r!Nc:||gdS)Nr) format_ticks)r;xposs rrBzStrCategoryFormatter.__call__s  !%%a((r!clfdjDfd|DS)NcBi|]\}}||Sr4)_text).0kvr;s r z5StrCategoryFormatter.format_ticks..s)FFF$!QQ 1 FFFr!cVg|]%}t|d&S))getround)rQval r_mappings r z5StrCategoryFormatter.format_ticks..s-@@@# eCjj"--@@@r!)r9items)r;rrZs` @rrKz!StrCategoryFormatter.format_tickssEFFFF$+2C2C2E2EFFF @@@@@@@@r!ct|tr|d}n$t|tst|}|S)z0Convert text values into utf-8 or ascii strings.zutf-8)encoding) isinstancebytesdecodestrrs rrPzStrCategoryFormatter._textsM eU # # LL'L22EEE3'' JJE r!r@) r0r1r2rGr=rBrKr3rPr4r!rr&r&sj::$$$))))AAA \r!r&c2eZdZddZedZdZdS)r+Nct|_tj|_|||dSdS)z Create mapping between unique categorical values and integer ids. Parameters ---------- data : iterable sequence of string values N)rr itertoolscount_counterr)r;r,s rr=zUnitData.__init__sD$  !))   KK       r!c t|nJ#t$r= tj|n#tt f$rYYdSwxYwYnwxYwdS)zY Helper method to check whether a string can be parsed as float or date. FT)rrdateutilparserparse TypeError)rYs r_str_is_convertiblezUnitData._str_is_convertibles   #JJJJ    %%c**** *   uuu +*  ts/ A=AAAAAActjtj|t}d}t j|D]`}t jttf||r| |}||j vrt|j |j |<a|jr|rtddSdSdS)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 TrczUsing 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_isinstancerbr`rnrnextrhsize_loginfo)r;r, convertiblerYs rrzUnitData.updates}RXd&999:: '-- 9 9C  !3,c : : : : <"66s;; $-''%)$-%8%8 c" 9 @ @ II? @ @ @ @ @ @ @ @ @r!r@)r0r1r2r=r3rnrr4r!rr+r+sW      \ @@@@@r!r+)rG collectionsrdateutil.parserrjrfloggingnumpyr matplotlibrrrr getLoggerr0rtConversionInterfacer Locatorr% Formatterr&r+registryrbstr_r`bytes_r4r!rrs  $#####111111111111w""ZHZHZHZHZH54ZHZHZHz(6+<:@:@:@:@:@:@:@:@D!5 4 6 67s7 N277 N5 N29r!