G8cdZddlmZddlZddlZddlZddlZddl 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)StrCategoryConverterc&| tdtj|tjtj |t }tj5td|D}ddd|jr4r2tjddtj|tS|j|tj|j j"tg|S#1swYxYw) 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 datadtypec3K|]<}tjj|xrt|tt f >ywN)rConversionInterface is_numlike isinstancestrbytes).0vs 5/usr/lib/python3/dist-packages/matplotlib/category.py z/StrCategoryConverter.convert..:sF.!"#66AA!DA%/C<%@!@A.sAAz3.5zSupport for passing numbers through unit converters is deprecated since %(since)s and support will be removed %(removal)s; use Axis.convert_units instead.)message)otypes) ValueErrorr_validate_unitnp atleast_1darrayobjectr'suppress_matplotlib_deprecation_warningallsizewarn_deprecatedasarrayfloatupdate vectorize_mapping __getitem__)valueunitaxisvaluesrs rconvertzStrCategoryConverter.converts* <   ++D1rxxV<=  9 9 ; ..&,..J . ;;:   G H::fE2 2 FFr||DMM55ugFvNN . .s %DDctj|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)rrStrCategoryLocatorr'StrCategoryFormatterrAxisInfo)r*r+r/r0s raxisinfozStrCategoryConverter.axisinfoGsA( ++D1$DMM2%dmm4~~VF;;c|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%)datar+s r default_unitsz"StrCategoryConverter.default_unitsbsH" ::  NN8D> *zz JJ  d #zzr5c:t|dstd|dy)Nr'zProvided unit "zU" is not valid for a categorical converter, as it does not have a _mapping attribute.)hasattrr)r*s rrz#StrCategoryConverter._validate_unitys3tZ(!$(GGH H)r5N)__name__ __module__ __qualname__ staticmethodr-r4r:rr5rrrsU)O)OV<<4,HHr5rc"eZdZdZdZdZdZy)r1z1Tick 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__ $ r5cHt|jjSr )listrFr,)rHs r__call__zStrCategoryLocator.__call__sDKK&&())r5c|Sr rA)rHvminvmaxs r tick_valueszStrCategoryLocator.tick_valuess v r5N)r=r>r?__doc__rJrNrRrAr5rr1r1s;$*r5r1c4eZdZdZdZddZdZedZy)r2z0String representation of the data at every tick.c||_yrDrErGs rrJzStrCategoryFormatter.__init__rKr5Nc,|j|gdS)Nr) format_ticks)rHxposs rrNzStrCategoryFormatter.__call__s  !%a((r5c|jjDcic]\}}||j|}}}|Dcgc]}|jt |dc}Scc}}wcc}w)N)rFitems_textgetround)rHr,kr r_mappingvals rrWz!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)rrdecoderr)s rr]zStrCategoryFormatter._texts? eU #LL'L2E E3'JE r5r ) r=r>r?rSrJrNrWr@r]rAr5rr2r2s):$)A r5r2c*eZdZddZedZdZy)r8Nc|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%)rHr9s rrJzUnitData.__init__s2$  !)   KK  r5c 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)r$rdateutilparserparse TypeError)rbs 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 TrfzUsing 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_isinstancerrrqr'nextrkr!_loginfo)rHr9 convertiblerbs rr%zUnitData.updates}}RXXd&9: ''- 9C  ! !3,c :"66s; $--'%)$--%8 c" 9 99 II? @%9r5r )r=r>r?rJr@rqr%rAr5rr8r8s!   @r5r8)rS collectionsrdateutil.parserrmriloggingnumpyr matplotlibrrr getLoggerr=rvrrLocatorr1 Formatterr2r8registryrstr_rbytes_rAr5rrs $**w"dH544dHN(6++<:@:@|+,s.0rww,.u02ryyr5