B Jd@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)_apitickerunitsc@s<eZdZeddZeddZeddZeddZd S) StrCategoryConverterc Cs|dkrtdt|ttj|td}tt dd|D}WdQRX|j rz|rztj dddtj |t dS||tj|jjt gd |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 data)dtypecss*|]"}tj|o t|ttf VqdS)N)rConversionInterface is_numlike isinstancestrbytes).0vrE/var/www/html/venv/lib/python3.7/site-packages/matplotlib/category.py :sz/StrCategoryConverter.convert..z3.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)Zotypes) ValueErrorr_validate_unitnp atleast_1darrayobjectrZ'suppress_matplotlib_deprecation_warningallsizeZwarn_deprecatedZasarrayfloatupdateZ vectorize_mapping __getitem__)valueunitaxisvaluesr rrrconverts     zStrCategoryConverter.convertcCs,t|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 .. note:: *axis* is not used Returns ------- `~matplotlib.units.AxisInfo` Information to support default tick labeling )majlocmajfmt)rrStrCategoryLocatorrStrCategoryFormatterrZAxisInfo)r r!r$r%rrraxisinfoGs   zStrCategoryConverter.axisinfocCs,|jdkr|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 N)rZ set_unitsUnitDatar)datar!rrr default_unitsbs  z"StrCategoryConverter.default_unitscCst|dstd|ddS)NrzProvided unit "zU" is not valid for a categorical converter, as it does not have a _mapping attribute.)hasattrr)r rrrrys z#StrCategoryConverter._validate_unitN)__name__ __module__ __qualname__ staticmethodr#r(r+rrrrrrs ,  rc@s(eZdZdZddZddZddZdS) r&z1Tick at every integer mapping of the string data.cCs ||_dS)z Parameters ---------- units_mapping : dict Mapping of category names (str) to indices (int). N)_units)self units_mappingrrr__init__szStrCategoryLocator.__init__cCst|jS)N)listr1r")r2rrr__call__szStrCategoryLocator.__call__cCs|S)Nr)r2ZvminZvmaxrrr tick_valuesszStrCategoryLocator.tick_valuesN)r-r.r/__doc__r4r6r7rrrrr&s r&c@s6eZdZdZddZd ddZddZed d ZdS) r'z0String representation of the data at every tick.cCs ||_dS)z Parameters ---------- units_mapping : dict Mapping of category names (str) to indices (int). N)r1)r2r3rrrr4szStrCategoryFormatter.__init__NcCs||gdS)Nr) format_ticks)r2xposrrrr6szStrCategoryFormatter.__call__cs*fddjDfdd|DS)Ncsi|]\}}||qSr)_text)r kr)r2rr sz5StrCategoryFormatter.format_ticks..csg|]}t|dqS))getround)r val) r_mappingrr sz5StrCategoryFormatter.format_ticks..)r1items)r2r"r)rCr2rr9sz!StrCategoryFormatter.format_tickscCs.t|tr|jdd}nt|ts*t|}|S)z0Convert text values into utf-8 or ascii strings.zutf-8)encoding)r r decoder )rrrrr<s   zStrCategoryFormatter._text)N) r-r.r/r8r4r6r9r0r<rrrrr's  r'c@s*eZdZdddZeddZddZdS) r)NcCs(t|_t|_|dk r$||dS)z Create mapping between unique categorical values and integer ids. Parameters ---------- data : iterable sequence of string values N)rr itertoolscount_counterr)r2r*rrrr4s  zUnitData.__init__c CsPy t|Wn>tk rJytj|Wnttfk rDdSXYnXdS)zY Helper method to check whether a string can be parsed as float or date. FT)rrdateutilparserparse TypeError)rBrrr_str_is_convertibles  zUnitData._str_is_convertiblecCs~ttj|td}d}xLt|D]>}tjtt f|d|rH| |}||j kr$t |j |j |<q$W|jrz|rztddS)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. )rT)rzUsing 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)rrrrrfromkeysrZcheck_isinstancer r rOrnextrJr_loginfo)r2r*Z convertiblerBrrrrs    zUnitData.update)N)r-r.r/r4r0rOrrrrrr)s  r))r8 collectionsrZdateutil.parserrKrHloggingnumpyrZ matplotlibrrr getLoggerr-rRrrZLocatorr& Formatterr'r)registryr Zstr_r bytes_rrrr s  g>