U
SZf%3 @ sz d dl Z d dlmZmZ d dlmZ d dlmZ d dlZd dl m
Z e dZ
dd Zdddddddd ddi f d
d ZdS ) N)
exceptionsoptional_imports)utilsZpandasc
C s t | }
|dkr|
nd}|dkr$dn|
}| s4d|
} |
s@d|
}
tjj||d| |
d}|d jtg ddd d
dd g t|dkrd nd
dd |d | |dkrd}d}nd}d}|d D ]z}d|ksd|krd|d | d< d|d | d< ||kr
d|d | d< ||krd|d | d< ddg|d | d< q|
dkrZddg|d |d d< |shd d g}|svd d g}t|
D ] }tt | j| d D ]}t |d |d t | j| d d!}|dkrt
| j| d d"| gndg}|dkr dgnt
| j| d d"| g}t j ||t|d"| d#d |dkrDd$nd%|d&dd'
|d d(
|d d)
}|| qtt | j| d* D ]}t |d |d t | j| d* d!}|dkrt
| j| d* d"| gnd+g}|dkrd+gnt
| j| d* d"| g}t j ||t|d"| d#d*|dkr8d$nd%|ddd'
|d d(
|d d)
}|| q|dkr| j| d, nd+g}|dkrd+gn | j| d, }t jf ||d,|dkrd$nd%d'
|d d(
|d d-| }|| | j| d. }d/| kr d0
| j| d/ }nd1}d2
|| } t| |dkrJ|
| n|d |
|dkr`|
n| |dkrpd3nd4|dkrd5ndd} |d d6 | f7 < q~|S )7Nh g ?F)Z
print_gridhorizontal_spacingvertical_spacingZlayout)Zshapesz Bullet ChartiX i stackx P )l)titleheightwidthZ
showlegendZbarmode annotationsmarginyaxisxaxisZshowgridZzerolineZ tickwidthZshowticklabelsr rangeg?g333333?1domainrgb(200, 200, 200)rgb(245, 245, 245)rgb(31, 119, 180)rgb(176, 196, 221)rangesrgb)colorxy zx{}zy{})
r r markername hoverinfo orientationr baser r measuresg ?markers)r r r# r$ r r titles subtitlesz
{} z {}rowcolTr )lenplotlyZsubplotsZ
make_subplotsupdatedictr ZilocclrsZn_colorssortedgoBarformatZ add_traceZScatterr Zannotation_dict_for_label) dfr( r' r r* r) r% range_colorsmeasure_colorsr r scatter_optionslayout_optionsZ num_of_lanesZ num_of_rowsZ num_of_colsfigZ
width_axisZ length_axiskeyr, idxZ inter_colorsr r barr
subtitlelabelZannot rB A/tmp/pip-unpacked-wheel-5ksk5baj/plotly/figure_factory/_bullet.py_bullet
s
rD r )r r )r r c s t s tdt| r4tdd | D sJtdnt| t jsJtddddd d
g}
t| r t jr~fd d | D n d
gt | rfdd | D n d
gt | r fdd | D n g gt | rfdd | D n g gt | rfdd | D n g gt | g|
d}nt| t jrt jrD|
n d
gt | rb|
n d
gt | r|
n g gt | r|
n g gt | r|
n g gt | g|
d}t j |}dD ]`}t || D ]L\}}z*t
|}|s|dkr"g || |< W n tk
r: Y nX qq||fD ]>}|rLt |dkrntdt| t|dd }qLdddd d i}| i kr| | n2|d D ](}|| d kr|d | | d |< qt| |||| |
| |
}|S ) a
**deprecated**, use instead the plotly.graph_objects trace
:class:`plotly.graph_objects.Indicator`.
:param (pd.DataFrame | list | tuple) data: either a list/tuple of
dictionaries or a pandas DataFrame.
:param (str) markers: the column name or dictionary key for the markers in
each subplot.
:param (str) measures: the column name or dictionary key for the measure
bars in each subplot. This bar usually represents the quantitative
measure of performance, usually a list of two values [a, b] and are
the blue bars in the foreground of each subplot by default.
:param (str) ranges: the column name or dictionary key for the qualitative
ranges of performance, usually a 3-item list [bad, okay, good]. They
correspond to the grey bars in the background of each chart.
:param (str) subtitles: the column name or dictionary key for the subtitle
of each subplot chart. The subplots are displayed right underneath
each title.
:param (str) titles: the column name or dictionary key for the main label
of each subplot chart.
:param (bool) orientation: if 'h', the bars are placed horizontally as
rows. If 'v' the bars are placed vertically in the chart.
:param (list) range_colors: a tuple of two colors between which all
the rectangles for the range are drawn. These rectangles are meant to
be qualitative indicators against which the marker and measure bars
are compared.
Default=('rgb(200, 200, 200)', 'rgb(245, 245, 245)')
:param (list) measure_colors: a tuple of two colors which is used to color
the thin quantitative bars in the bullet chart.
Default=('rgb(31, 119, 180)', 'rgb(176, 196, 221)')
:param (float) horizontal_spacing: see the 'horizontal_spacing' param in
plotly.tools.make_subplots. Ranges between 0 and 1.
:param (float) vertical_spacing: see the 'vertical_spacing' param in
plotly.tools.make_subplots. Ranges between 0 and 1.
:param (dict) scatter_options: describes attributes for the scatter trace
in each subplot such as name and marker size. Call
help(plotly.graph_objs.Scatter) for more information on valid params.
:param layout_options: describes attributes for the layout of the figure
such as title, height and width. Call help(plotly.graph_objs.Layout)
for more information on valid params.
Example 1: Use a Dictionary
>>> import plotly.figure_factory as ff
>>> data = [
... {"label": "revenue", "sublabel": "us$, in thousands",
... "range": [150, 225, 300], "performance": [220,270], "point": [250]},
... {"label": "Profit", "sublabel": "%", "range": [20, 25, 30],
... "performance": [21, 23], "point": [26]},
... {"label": "Order Size", "sublabel":"US$, average","range": [350, 500, 600],
... "performance": [100,320],"point": [550]},
... {"label": "New Customers", "sublabel": "count", "range": [1400, 2000, 2500],
... "performance": [1000, 1650],"point": [2100]},
... {"label": "Satisfaction", "sublabel": "out of 5","range": [3.5, 4.25, 5],
... "performance": [3.2, 4.7], "point": [4.4]}
... ]
>>> fig = ff.create_bullet(
... data, titles='label', subtitles='sublabel', markers='point',
... measures='performance', ranges='range', orientation='h',
... title='my simple bullet chart'
... )
>>> fig.show()
Example 2: Use a DataFrame with Custom Colors
>>> import plotly.figure_factory as ff
>>> import pandas as pd
>>> data = pd.read_json('https://cdn.rawgit.com/plotly/datasets/master/BulletData.json')
>>> fig = ff.create_bullet(
... data, titles='title', markers='markers', measures='measures',
... orientation='v', measure_colors=['rgb(14, 52, 75)', 'rgb(31, 141, 127)'],
... scatter_options={'marker': {'symbol': 'circle'}}, width=700)
>>> fig.show()
z3'pandas' must be installed for this figure factory.c s s | ]}t |tV qd S )N)
isinstancer1 ).0itemrB rB rC s z create_bullet..zGEvery entry of the data argument list, tuple, etc must be a dictionary.z=You must input a pandas DataFrame, or a list of dictionaries.r) r@ r( r' r c s g | ] }| qS rB rB rF d)r) rB rC
' s z!create_bullet..r+ c s g | ] }| qS rB rB rI )r* rB rC rK ( s c s g | ] }| qS rB rB rI )r( rB rC rK ) s c s g | ] }| qS rB rB rI )r' rB rC rK * s c s g | ] }| qS rB rB rI )r rB rC rK + s )index)r r' r( Nr! zKBoth 'range_colors' or 'measure_colors' must be a list of two valid colors.r r r" z diamond-tallz rgb(0, 0, 0))sizesymbolr )pd ImportErrorr Z is_sequenceallr Z PlotlyErrorrE Z DataFramer. tolistZ transpose enumeratemathisnan TypeErrorr2 Zvalidate_colorsZconvert_colors_to_same_typer0 rD )datar( r' r r* r) r% r8 r9 r r r: r; Z col_namesr7 Z
needed_keyr> rZr_is_nanZ colors_listZdefault_scatterkr< rB )r( r' r r* r) rC
create_bullet s ]
""""$
r[ )rU r/ r r Z
plotly.colorscolorsr2 Zplotly.figure_factoryr Zplotly.graph_objsZ
graph_objsr4 Z
get_modulerP rD r[ rB rB rB rC s(
,