gdZddlmZddlZddlZddlZddlZddl m Z m Z m Z ej eZGdde jZGdde jZGd d e jZGd 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)_apitickerunitscfeZdZedZedZedZedZdS)StrCategoryConverterc:|tdt|tjtj|t }||tj|j j tg|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 NzMissing 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 I/granary/edev/quantum/lib/python3.11/site-packages/matplotlib/category.pyconvertzStrCategoryConverter.converts* <    ++D111rxV<<<== FFr|DM5ugFFFvNNNct|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<sM( ++D111$DM22%dm44~VF;;;;rc|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_unitsWsE" :  NN8D>> * * * * J  d # # #zrcLt|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_unitnsPtZ(( HG$GGGHH H H HrN)__name__ __module__ __qualname__ staticmethodrr%r*rrrrrsOO\O@<<\<4\,HH\HHHrrc$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__x$ rcNt|jSN)listr6r)r8s r__call__zStrCategoryLocator.__call__sDK&&(()))rc|Sr=r1)r8vminvmaxs r tick_valueszStrCategoryLocator.tick_valuess tvv rN)r-r.r/__doc__r:r?rCr1rrr"r"vsG;;$$$***rr"c<eZdZdZdZddZdZedZdS)r#z0String representation of the data at every tick.c||_dSr4r5r7s rr:zStrCategoryFormatter.__init__r;rNc:||gdS)Nr) format_ticks)r8xposs rr?zStrCategoryFormatter.__call__s  !%%a((rclfdjDfd|DS)NcBi|]\}}||Sr1)_text).0kvr8s r z5StrCategoryFormatter.format_ticks..s)FFF$!QQ 1 FFFrcVg|]%}t|d&S))getround)rNval r_mappings r z5StrCategoryFormatter.format_ticks..s-@@@# eCjj"--@@@r)r6items)r8rrWs` @rrHz!StrCategoryFormatter.format_tickssEFFFF$+2C2C2E2EFFF @@@@@@@@rct|tr|d}n$t|tst|}|S)z0Convert text values into utf-8 or ascii strings.zutf-8)encoding) isinstancebytesdecodestrrs rrMzStrCategoryFormatter._textsM eU # # LL'L22EEE3'' JJE rr=) r-r.r/rDr:r?rHr0rMr1rrr#r#sj::$$$))))AAA \rr#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)r8r)s rr:zUnitData.__init__sD$  !))   KK       rc 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)rr dateutilparserparse TypeError)rVs 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 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_isinstancer_r]rkrnextresize_loginfo)r8r) convertiblerVs rrzUnitData.updates}RXd&999:: '-- 9 9C  !3,c : : : : <"66s;; $-''%)$-%8%8 c" 9 @ @ II? @ @ @ @ @ @ @ @ @rr=)r-r.r/r:r0rkrr1rrr(r(sW      \ @@@@@rr()rD collectionsrdateutil.parserrgrcloggingnumpyr matplotlibrrr getLoggerr-rqConversionInterfacerLocatorr" Formatterr#r(registryr_str_r]bytes_r1rrrs  $#####**********w""YHYHYHYHYH54YHYHYHx(6+<:@:@:@:@:@:@:@:@|+*,,s..00rw,,..u0022ryr