OwlCyberSecurity - MANAGER
Edit File: category.cpython-38.pyc
U ��]E � @ s� d Z ddlmZ ddlZddlZddlZddlZddl m Z ddl m Z ddl mZ e�e�ZG dd� de j�ZG dd� dej�ZG dd � d ej�ZG d d � d e�Ze� e je< e� e jej < e� e je < e� e jej < dS ) a� Module that allows plotting of string "category" data. i.e. ``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, provides a tick locator and formatter, and the class:`.UnitData` that creates and stores the string-to-integer mapping. � )� OrderedDictNc @ s0 e Zd Zedd� �Zedd� �Zedd� �ZdS )�StrCategoryConverterc C sn |dkrt d��t�tj| td��}tdd� |D ��rDtj|td�S |�|� tj |j j tgd�}||�}|S )a Convert strings in value to floats using mapping information store in the unit object. Parameters ---------- value : string or iterable Value or list of values to be converted. unit : `.UnitData` An object mapping strings to integers. axis : `~matplotlib.axis.Axis` axis on which the converted value is plotted. .. note:: *axis* is unused. Returns ------- mapped_value : float or ndarray[float] NzMissing category information for StrCategoryConverter; this might be caused by unintendedly mixing categorical and numeric data�Zdtypec s s* | ]"}t j�|�o t|ttf� V qd S �N)�units�ConversionInterfaceZ is_numlike� isinstance�str�bytes)�.0�v� r �5/usr/lib/python3/dist-packages/matplotlib/category.py� <genexpr>< s � z/StrCategoryConverter.convert.<locals>.<genexpr>)Zotypes) � ValueError�np� atleast_1d�array�object�allZasarray�float�updateZ vectorize�_mapping� __getitem__)�value�unit�axis�valuesZstr2idxZ mapped_valuer r r �convert s �� �z StrCategoryConverter.convertc C s" t | j�}t| j�}tj||d�S )a� Sets 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 ------- axisinfo : `~matplotlib.units.AxisInfo` Information to support default tick labeling .. note: axis is not used )�majloc�majfmt)�StrCategoryLocatorr �StrCategoryFormatterr ZAxisInfo)r r r r r r r �axisinfoI s z StrCategoryConverter.axisinfoc C s, |j dkr|�t| �� n |j �| � |j S )a^ Sets and updates the :class:`~matplotlib.Axis.axis` units. Parameters ---------- data : string or iterable of strings axis : `~matplotlib.Axis.axis` axis on which the data is plotted Returns ------- class : `.UnitData` object storing string to integer mapping N)r Z set_units�UnitDatar )�datar r r r � default_unitsa s z"StrCategoryConverter.default_unitsN)�__name__� __module__� __qualname__� staticmethodr r# r& r r r r r s * r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) r! z0tick at every integer mapping of the string datac C s || _ dS )z{ Parameters ----------- units_mapping : Dict[str, int] string:integer mapping N��_units��selfZ units_mappingr r r �__init__{ s zStrCategoryLocator.__init__c C s t | j�� �S r )�listr, r )r. r r r �__call__� s zStrCategoryLocator.__call__c C s | � S r r )r. ZvminZvmaxr r r � tick_values� s z StrCategoryLocator.tick_valuesN)r' r( r) �__doc__r/ r1 r2 r r r r r! y s r! c @ s. e Zd ZdZdd� Zd dd�Zedd� �ZdS ) r" z/String representation of the data at every tickc C s || _ dS )zy Parameters ---------- units_mapping : Dict[Str, int] string:integer mapping Nr+ r- r r r r/ � s z StrCategoryFormatter.__init__Nc C s6 |d kr dS dd� | j �� D �}|�tt�|��d�S )N� c S s i | ]\}}|t �|��qS r )r" �_text)r �kr r r r � <dictcomp>� s � z1StrCategoryFormatter.__call__.<locals>.<dictcomp>)r, �items�get�intr �round)r. �x�posZ r_mappingr r r r1 � s �z StrCategoryFormatter.__call__c C s. t | t�r| jdd�} nt | t�s*t| �} | S )z:Converts text values into utf-8 or ascii strings. zutf-8)�encoding)r r �decoder )r r r r r5 � s zStrCategoryFormatter._text)N)r' r( r) r3 r/ r1 r* r5 r r r r r" � s r" c @ s* e Zd Zddd�Zedd� �Zdd� ZdS ) r$ Nc C s( t � | _t�� | _|dk r$| �|� dS )z� Create mapping between unique categorical values and integer ids. Parameters ---------- data : iterable sequence of string values N)r r � itertools�count�_counterr )r. r% r r r r/ � s zUnitData.__init__c C sP z t | � W n> tk rJ ztj�| � W n tk rD Y Y dS X Y nX dS )zb Helper method to see if a string can be cast to float or parsed as date. FT)r r �dateutil�parser�parse��valr r r �_str_is_convertible� s z UnitData._str_is_convertiblec C s� t �t j|td��}d}t�|�D ]J}t|ttf�sDt dj |d���|rR| � |�}|| j kr"t | j�| j |< q"|r|t�d� dS )a Maps new values to integer identifiers. Parameters ---------- data : iterable sequence of string values Raises ------ TypeError If the value in data is not a string, unicode, bytes type r Tz{val!r} is not a stringrF z�Using 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)r r r r r �fromkeysr r r � TypeError�formatrH r �nextrB �_log�info)r. r% Z convertiblerG r r r r � s zUnitData.update)N)r' r( r) r/ r* rH r r r r r r$ � s r$ ) r3 � collectionsr Zdateutil.parserrC r@ ZloggingZnumpyr Zmatplotlib.cbookZcbookZmatplotlib.unitsr Zmatplotlib.tickerZtickerZ getLoggerr' rM r r ZLocatorr! Z Formatterr" r r$ �registryr Zstr_r Zbytes_r r r r �<module> s" \ @