|
Server : Apache System : Linux server112.speedpacket.com 3.10.0-962.3.2.lve1.5.36.el7.x86_64 #1 SMP Mon May 18 02:16:06 EDT 2020 x86_64 User : equistyl ( 1341) PHP Version : 8.1.30 Disable Function : NONE Directory : /lib64/python2.7/site-packages/matplotlib/ |
Upload File : |
�
r`]c @ s� d Z d Z d g Z d Z d Z d d l Z d d l m Z m Z d d l
m Z d d l m
Z
d d
l m Z d d l m Z d d l m Z d
Z d Z d Z d d d � � YZ d S( s6
Module for creating Sankey diagrams using matplotlib
s Kevin L. Daviess
Yannick Copint BSDs
2011/09/16i����N( t iterablet Bunch( t Path( t PathPatch( t Affine2D( t verbose( t docstringi i i t Sankeyc B s� e Z d Z d e d d d f d � Z d � Z d � Z e j d � Z e
j d e e d d d e d d f d d
� � Z
d � Z e d d d d d
d d d d d d � Z RS( sn
Sankey diagram in matplotlib
Sankey diagrams are a specific type of flow diagram, in which
the width of the arrows is shown proportionally to the flow
quantity. They are typically used to visualize energy or
material or cost transfers between processes.
`Wikipedia (6/1/2011) <http://en.wikipedia.org/wiki/Sankey_diagram>`_
i i c C s� t j t j t j t j t j t j t j g } t j d d g d d g d d g d d g d d g d d g d d g g � } | d k s� | d k r� | r� | } qW| d d � d d d
� f } n� | d k s� | d k rW| r t j | d d � d f | d d � d f f � } qWt j | d d � d f | d d � d f f � } n | d k rm| } n t | | | t j | | j d d f � � S(
s=
Return the codes and vertices for a rotated, scaled, and translated
90 degree arc.
Optional keyword arguments:
=============== ==========================================
Keyword Description
=============== ==========================================
*quadrant* uses 0-based indexing (0, 1, 2, or 3)
*cw* if True, clockwise
*center* (x, y) tuple of the arc's center
=============== ==========================================
g �?g gw,���?g����S��?g�����?g>�ef���?i i Ni����i i ( R t LINETOt CURVE4t npt arrayt column_stackt zipt tilet shape( t selft quadrantt cwt radiust centert ARC_CODESt ARC_VERTICESt vertices( ( s7 /usr/lib64/python2.7/site-packages/matplotlib/sankey.pyt _arc@ s8 "
c
C s� | d
k r" d d g d d g f S| d d \ } } | d | j } | t k r� | | 8} | | | | d g } | j t j | | g f t j | f t j | | | g f t j | | j | | g f g � | d | j | d g } n�| | j 8} | t k rd }
n d }
| | d | |
| | g } | t k rNd } n d } | j
r�| j | j d | d | t k d | j
d | | j
| |
| j
f � � n | j t j | | g f � | j t j | | |
| g f t j | f t j | | | |
| g f g � | j | j d | d | t k d | | j
d | | j
| |
| j
f � � | j t j | | | |
| g f � | d | d |
| j g } | | f Sd
S( sP
Add an input to a path and return its tip and label locations.
i i����i i g @R R R R N(
t Nonet pitcht RIGHTt extendR R t gapt offsett UPt DOWNR R t append( R t patht anglet flowt lengtht xt yt dipdeptht dipt label_locationt signR ( ( s7 /usr/lib64/python2.7/site-packages/matplotlib/sankey.pyt
_add_inputw sL
$
%
( c
C sA | d k r" d d g d d g f S| d d \ } } | j | d | j } | t k r.| | 7} | | | | d g } | j t j | | g f t j | | | j g f t j | f t j | | | j | g f t j | | | g f t j | | j | | g f g � | d | j | d g } n| | j 7} | t k rPd }
n d }
| | d | |
| | g } | t k r�d } n d } | j
r�| j | j d | d | t k d | j
d
| | j
| |
| j
f � � n | j t j | | g f � | j t j | | |
| g f t j | | j | |
| g f t j | f t j | | j | | |
| g f t j | | | |
| g f g � | j | j d | d | t
k d | j
| d
| | j
| |
| j
f � � | j t j | | | |
| g f � | d | d |
| j g } | | f Sd S( s�
Append an output to a path and return its tip and label locations.
Note: *flow* is negative for an output.
i i����i i g @i R R R R N( R t shoulderR R R R R R R R R R R" R! ( R R# R$ R% R& R' R( t tipheightt tipR+ R, R ( ( s7 /usr/lib64/python2.7/site-packages/matplotlib/sankey.pyt _add_output� sX
$
%
( c C sM g } | } x: | d d d � D]% \ } } | j | | f � | } q W| S( s�
A path is not simply revertable by path[::-1] since the code
specifies an action to take from the **previous** point.
Ni����( R" ( R R# t first_actiont reverse_patht next_codet codet position( ( s7 /usr/lib64/python2.7/site-packages/matplotlib/sankey.pyt _revert� s
t g �?g �?c
7 K sB | d k r$ t j d d g � } n t j | � } | j d } | d k rU d } n
| d :} | d k rz d d g } n | d k r� t | d t � r� n
| g | } t j t j | � � | j k r� t j
d t j | � d � n | j | } t d � | D� � }
t d
� | D� � } d |
k o9d k n sUt j
d
|
d � n d | k old k n s�t j
d |
d � n | d k r�| j | j
| d | | d } n d g | } xs t | � D]e \ } } | | j k r�t | | <q�| | j k rt | | <q�t j
d | | | j f d � q�Wd g | } x� t t | | � � D]� \ } \ } } | d k r�| r�t | | <q| t k rt | | <qqa| d k r�| d k rt | | <qqa| r�t | | <qa| t k rat | | <qaqaWt | � rn^| } | } | } | } t d | � } g | D] } | j | d � ^ qC} x� t t | | | � � D]h \ } \ } } } | t k r�| r�| | | <| | 7} qz| t k rz| rz| | | <| | 8} qzqzWx� t t t | | | � � � D]x \ } \ } } } | t k rK| rK| | | | d <| | 7} q| t k r| r| | | | d <| | 8} qqWt } xw t t t | | t | | � � � � D]N \ } \ } } } | t k r�| r�| r�d | | | d <q�t } q�q�q�Wt } xi t t | | t | | � � � D]F \ } \ } } } | t k r)| so| rcd | | <qlt } qoq)q)Wt j | j | d |
d g f t j | j | d d |
d g f t j | j | d d |
d g f t j | d | j d | d g f t j | d | j d | d g f t j | d | j | d g f g } t j | d | j | d g f t j | d | j d | d g f t j | d | j d | d g f t j | j | d d |
d g f t j | j | d d |
d g f t j | j | d |
d g f g } t j | d | j | d g f g }! t j | j | d |
d g f g }" t j | d f � }# t j | d f � }$ x� t t | | t | | � � � D]� \ } \ } } } | t k r0| r0| j |" | | � \ |# | d d � f <|$ | d d � f <q�| t k r�| r�| j | | | � \ |# | d d � f <|$ | d d � f <q�q�Wx� t t t | | t | | � � � � D]� \ } \ } } } | t k r" | r" | j | | | � \ |# | | d d d � f <|$ | | d d d � f <q�| t k r�| r�| j |! | | � \ |# | | d d d � f <|$ | | d d d � f <q�q�Wt } xt t t | | t | | � � � � D]� \ } \ } } } | t k r� | r� | sF
| d d d |" d d d k r=
| j t j |" d d d | d d d g f � n t } n | j | | | � \ |# | | d d d � f <|$ | | d d d � f <q� q� Wt } x� t t | | t | | � � � D]� \ } \ } } } | t k r�
| r�
| sR | d d d |! d d d k rI | j t j |! d d d | d d d g f � n t } n | j | | | � \ |# | d d � f <|$ | d d � f <q�
q�
W| s� |" j! � | j! � n | s� |! j! � | j! � n | | j" |! � | | j" |" � t j# | d d f g }% t |% � \ }& }' t j |' � }' d � }( | d k r� | d k r� g | D] } |( | | � ^ qS } t$ � j% | d � j& }) |) |# � }# |) |$ � }$ |) |' � }' n | j' j( d d d | d d d d �}* n| j | j) | d | | d } g | D] } |( | | � ^ q
} t$ � j% | d � j& }) |) |# � }# | j | j* | d |# | d }+ t$ � j+ |+ � j& }, |, |# � }# |, |) |$ � � }$ |, |) |' � � }' t d | d d d d � }- | j' j( |+ |- � }* t rCd G| GHd G| j" |" � GHd G| GHd! G| j" |! � GHt |' � \ }. }/ | j' j, |. |/ d" � n t- t |' |& � d# |
j! d# |
j! d$ d% � � d& |
j! d& |
j! d' d � � |
�}0 | j' j. |0 � x� t t | | � � D]� \ } \ }1 } | | d k s�| d k r�d | | <q�| j/ d k r�| j0 t1 |1 � | j/ }2 | | d k rD| | c d( 7<n | | c |2 7<q�q�Wg }3 xz t t | |$ � � D]c \ } \ }4 }5 |4 r�|4 }6 n d }6 |3 j | j' j( d) |5 d d* |5 d d |6 d d d d � � qwWt2 t j2 |' d d � d f � t j2 |$ d d � d f � | j3 d � t4 t j4 |' d d � d f � t j4 |$ d d � d f � | j3 d � t2 t j2 |' d d � d f � t j2 |$ d d � d f � | j3 d � t4 t j4 |' d d � d f � t j4 |$ d d � d f � | j3 d+ � f | _3 | j j t5 d, |0 d- | d. | d/ |# d0 |* d1 |3 � � | S(2 s
Add a simple Sankey diagram with flows at the same hierarchical level.
Return value is the instance of :class:`Sankey`.
Optional keyword arguments:
=============== ===================================================
Keyword Description
=============== ===================================================
*patchlabel* label to be placed at the center of the diagram
Note: *label* (not *patchlabel*) will be passed to
the patch through ``**kwargs`` and can be used to
create an entry in the legend.
*flows* array of flow values
By convention, inputs are positive and outputs are
negative.
*orientations* list of orientations of the paths
Valid values are 1 (from/to the top), 0 (from/to
the left or right), or -1 (from/to the bottom). If
*orientations* == 0, inputs will break in from the
left and outputs will break away to the right.
*labels* list of specifications of the labels for the flows
Each value may be None (no labels), '' (just label
the quantities), or a labeling string. If a single
value is provided, it will be applied to all flows.
If an entry is a non-empty string, then the
quantity for the corresponding flow will be shown
below the string. However, if the *unit* of the
main diagram is None, then quantities are never
shown, regardless of the value of this argument.
*trunklength* length between the bases of the input and output
groups
*pathlengths* list of lengths of the arrows before break-in or
after break-away
If a single value is given, then it will be applied
to the first (inside) paths on the top and bottom,
and the length of all other arrows will be
justified accordingly. The *pathlengths* are not
applied to the horizontal inputs and outputs.
*prior* index of the prior diagram to which this diagram
should be connected
*connect* a (prior, this) tuple indexing the flow of the
prior diagram and the flow of this diagram which
should be connected
If this is the first diagram or *prior* is None,
*connect* will be ignored.
*rotation* angle of rotation of the diagram [deg]
*rotation* is ignored if this diagram is connected
to an existing one (using *prior* and *connect*).
The interpretation of the *orientations* argument
will be rotated accordingly (e.g., if *rotation*
== 90, an *orientations* entry of 1 means to/from
the left).
=============== ===================================================
Valid kwargs are :meth:`matplotlib.patches.PathPatch` arguments:
%(Patch)s
As examples, ``fill=False`` and ``label='A legend entry'``.
By default, ``facecolor='#bfd1d4'`` (light blue) and
``linewidth=0.5``.
The indexing parameters (*prior* and *connect*) are zero-based.
The flows are placed along the top of the diagram from the inside out
in order of their index within the *flows* list or array. They are
placed along the sides of the diagram from the top down and along the
bottom from the outside in.
If the the sum of the inputs and outputs is nonzero, the discrepancy
will appear as a cubic Bezier curve along the top and bottom edges of
the trunk.
.. seealso::
:meth:`finish`
g �?g �i g �V@R8 t __iter__sH The sum of the flows is nonzero (%f).
Is the system not at steady state?t helpfulc s s | ] } t | d � Vq d S( i N( t max( t .0R% ( ( s7 /usr/lib64/python2.7/site-packages/matplotlib/sankey.pys <genexpr>l s c s s | ] } t | d � Vq d S( i N( t min( R< R% ( ( s7 /usr/lib64/python2.7/site-packages/matplotlib/sankey.pys <genexpr>m s g �?g @s� The scaled sum of the inputs is %f.
This may cause poor layout.
Consider changing the scale so that the scaled sum is approximately 1.0.g �g �s� The scaled sum of the outputs is %f.
This may cause poor layout.
Consider changing the scale so that the scaled sum is approximately 1.0.i sw The magnitude of flow %d (%f) is below the tolerance (%f).
It will not be shown, and it cannot be used in a connection.R g @i Ni����c S s | d k r d S| | Sd S( N( R ( t at r( ( s7 /usr/lib64/python2.7/site-packages/matplotlib/sankey.pyt
_get_angleM s iZ t st haR t vas llpath
s ulpath
s urpath
s lrpath
s go-t fct facecolors #bfd1d4t lwt linewidths
R' R( i t patcht flowst anglest tipst textt texts(6 R R R R t getattrt Falset absolutet sumt toleranceR t reportt scalet diagramsRI t enumeratet TrueR R! R R R t dictt gett reversedR t MOVETOR R R
t zerosR- R1 R" t popR7 t CLOSEPOLYR t
rotate_degt transform_pointt axRL RJ RK t translatet plotR t add_patcht unitt formatt absR= t extentR; R (7 R t
patchlabelRI t orientationst labelst trunklengtht pathlengthst priort connectt rotationt kwargst nt scaled_flowst gaint losst
flow_errort
are_inputst iR% RJ t orientt is_inputt urlengtht ullengtht lrlengtht lllengtht dR$ t has_left_inputt spect has_right_outputt urpatht llpatht lrpatht ulpathRK t label_locationsR# t codesR R@ t rotateRL R Rb t kwdst xst ysRH t numbert quantityRM t labelt locationRA ( ( s7 /usr/lib64/python2.7/site-packages/matplotlib/sankey.pyt add� s� T
!
(
%
(
1 .
.8< 1HL 1$ L .$ <
! " *" $ (
( &
!c C sq | j j | j d | j | j d | j | j d | j | j d | j g � | j j d d d �| j S( s�
Adjust the axes and return a list of information about the Sankey
subdiagram(s).
Return value is a list of subdiagrams represented with the following
fields:
=============== ===================================================
Field Description
=============== ===================================================
*patch* Sankey outline (an instance of
:class:`~maplotlib.patches.PathPatch`)
*flows* values of the flows (positive for input, negative
for output)
*angles* list of angles of the arrows [deg/90]
For example, if the diagram has not been rotated,
an input to the top side will have an angle of 3
(DOWN), and an output from the top side will have
an angle of 1 (UP). If a flow has been skipped
(because its magnitude is less than *tolerance*),
then its angle will be None.
*tips* array in which each row is an [x, y] pair
indicating the positions of the tips (or "dips") of
the flow paths
If the magnitude of a flow is less the *tolerance*
for the instance of :class:`Sankey`, the flow is
skipped and its tip will be at the center of the
diagram.
*text* :class:`~matplotlib.text.Text` instance for the
label of the diagram
*texts* list of :class:`~matplotlib.text.Text` instances
for the labels of flows
=============== ===================================================
.. seealso::
:meth:`add`
i i i i t equalt
adjustablet datalim( Ra t axisRh t margint
set_aspectRU ( R ( ( s7 /usr/lib64/python2.7/site-packages/matplotlib/sankey.pyt finish� s 's %Gg�������?g� ��Q��?g333333�?id g�������?g���ư>c K s | d k rK d d l j }
|
j � } | j d d d d g d g �} n g | _ | | _ | | _ | | _ | | _ | | _
| | _ | | _ | | _
|
| _ t j t j d | d d � | _ | | _ t j t j t j t j t j f � | _ t | � r| j | � n d S( s=
Create a new Sankey instance.
Optional keyword arguments:
=============== ===================================================
Field Description
=============== ===================================================
*ax* axes onto which the data should be plotted
If *ax* isn't provided, new axes will be created.
*scale* scaling factor for the flows
*scale* sizes the width of the paths in order to
maintain proper layout. The same scale is applied
to all subdiagrams. The value should be chosen
such that the product of the scale and the sum of
the inputs is approximately 1.0 (and the product of
the scale and the sum of the outputs is
approximately -1.0).
*unit* string representing the physical unit associated
with the flow quantities
If *unit* is None, then none of the quantities are
labeled.
*format* a Python number formatting string to be used in
labeling the flow as a quantity (i.e., a number
times a unit, where the unit is given)
*gap* space between paths that break in/break away
to/from the top or bottom
*radius* inner radius of the vertical paths
*shoulder* size of the shoulders of output arrowS
*offset* text offset (from the dip or tip of the arrow)
*head_angle* angle of the arrow heads (and negative of the angle
of the tails) [deg]
*margin* minimum space between Sankey outlines and the edge
of the plot area
*tolerance* acceptable maximum of the magnitude of the sum of
flows
The magnitude of the sum of connected flows cannot
be greater than *tolerance*.
=============== ===================================================
The optional arguments listed above are applied to all subdiagrams so
that there is consistent alignment and formatting.
If :class:`Sankey` is instantiated with any keyword arguments other
than those explicitly listed above (``**kwargs``), they will be passed
to :meth:`add`, which will create the first subdiagram.
In order to draw a complex Sankey diagram, create an instance of
:class:`Sankey` by calling it without any kwargs::
sankey = Sankey()
Then add simple Sankey sub-diagrams::
sankey.add() # 1
sankey.add() # 2
#...
sankey.add() # n
Finally, create the full diagram::
sankey.finish()
Or, instead, simply daisy-chain those calls::
Sankey().add().add... .add().finish()
.. seealso::
:meth:`add`
:meth:`finish`
**Examples:**
.. plot:: mpl_examples/api/sankey_demo_basics.py
i����Ni t xtickst yticksg �f@g @( R t matplotlib.pyplott pyplott figuret add_subplotRU Ra Re Rf RT R R R. R R� R t tant piR RR R t infRh t lenR� ( R Ra RT Re Rf R R R. R t
head_angleR� RR Rq t pltt fig( ( s7 /usr/lib64/python2.7/site-packages/matplotlib/sankey.pyt __init__� s& _ $ % , ( t __name__t
__module__t __doc__RW R R- R1 R R R7 R t dedent_interpdR R� R� R� ( ( ( s7 /usr/lib64/python2.7/site-packages/matplotlib/sankey.pyR 5 s
7 4 : � � . ( ( R� t
__author__t __credits__t __license__t __version__t numpyR t matplotlib.cbookR R t matplotlib.pathR t matplotlib.patchesR t matplotlib.transformsR t
matplotlibR R R R R! R ( ( ( s7 /usr/lib64/python2.7/site-packages/matplotlib/sankey.pyt <module> s