B
eJd$ @ sX d dl mZ d dlmZmZ d dlmZ edZdd Z dd
dZ
G dd de Z dS ) )absolute_import)
exceptionsoptional_imports)
graph_objsZpandasc C s( ddt | g}t ||kr$tddS )z
Table-specific validations
Check that font_colors is supplied correctly (1, 3, or len(text)
colors).
:raises: (PlotlyError) If font_colors is supplied incorretly.
See FigureFactory.create_table() for params
z>Oops, font_colors should be a list of length 1, 3 or len(text)N)lenr Z PlotlyError)
table_text font_colorsZfont_colors_len_options r N/var/www/html/venv/lib/python3.7/site-packages/plotly/figure_factory/_table.pyvalidate_table s r
NF ? nonec
K s |dk r |nddgddgddgg}|dk r.|ndddg}t | | t| |||||f| } t| |||||f| }
tf d | d
|d |d |} | g} t|
t| | d
tdddddtdd ddddd dtd ddddd dd}
tj| |
dS )a
Function that creates data tables.
See also the plotly.graph_objects trace
:class:`plotly.graph_objects.Table`
:param (pandas.Dataframe | list[list]) text: data for table.
:param (str|list[list]) colorscale: Colorscale for table where the
color at value 0 is the header color, .5 is the first table color
and 1 is the second table color. (Set .5 and 1 to avoid the striped
table effect). Default=[[0, '#66b2ff'], [.5, '#d9d9d9'],
[1, '#ffffff']]
:param (list) font_colors: Color for fonts in table. Can be a single
color, three colors, or a color for each row in the table.
Default=['#000000'] (black text for the entire table)
:param (int) height_constant: Constant multiplied by # of rows to
create table height. Default=30.
:param (bool) index: Create (header-colored) index column index from
Pandas dataframe or list[0] for each list in text. Default=False.
:param (string) index_title: Title for index column. Default=''.
:param kwargs: kwargs passed through plotly.graph_objs.Heatmap.
These kwargs describe other attributes about the annotated Heatmap
trace such as the colorscale. For more information on valid kwargs
call help(plotly.graph_objs.Heatmap)
Example 1: Simple Plotly Table
>>> from plotly.figure_factory import create_table
>>> text = [['Country', 'Year', 'Population'],
... ['US', 2000, 282200000],
... ['Canada', 2000, 27790000],
... ['US', 2010, 309000000],
... ['Canada', 2010, 34000000]]
>>> table = create_table(text)
>>> table.show()
Example 2: Table with Custom Coloring
>>> from plotly.figure_factory import create_table
>>> text = [['Country', 'Year', 'Population'],
... ['US', 2000, 282200000],
... ['Canada', 2000, 27790000],
... ['US', 2010, 309000000],
... ['Canada', 2010, 34000000]]
>>> table = create_table(text,
... colorscale=[[0, '#000000'],
... [.5, '#80beff'],
... [1, '#cce5ff']],
... font_colors=['#ffffff', '#000000',
... '#000000'])
>>> table.show()
Example 3: Simple Plotly Table with Pandas
>>> from plotly.figure_factory import create_table
>>> import pandas as pd
>>> df = pd.read_csv('http://www.stat.ubc.ca/~jenny/notOcto/STAT545A/examples/gapminder/data/gapminderDataFiveYear.txt', sep=' ')
>>> df_p = df[0:25]
>>> table_simple = create_table(df_p)
>>> table_simple.show()
Nr z#00083eg ?z#ededeer z#ffffffz#000000Zheatmapg ?F)typezZopacity
colorscaleZ showscale hoverinfo2 )tbrlreversed r )Z autorangezeroline gridwidthticksdticktick0showticklabelsg )r r r r r! r" ) annotationsheightmarginZyaxisZxaxis)datalayout)r
_Tableget_table_matrixmake_table_annotationsdictr r ZFigure)r r r
index index_titleannotation_offsetZheight_constantr kwargs table_matrixr# tracer& r' r r r create_table sb O
r2 c @ s0 e Zd ZdZdd Zdd Zdd Zdd Zd
S ) r( z<
Refer to TraceFactory.create_table() for docstring
c K s t rtt|t jrt|j }|j } |j }|d| |rt| d| x(tt |D ]}
||
d| |
qXW || _
|| _ || _ || _|| _
tt |d | _tt || _d S )Nr )pd
isinstanceZ DataFramecolumnstolistr, valuesinsertranger r r r
r. xy) selfr r r
r, r- r. r/ headersZtable_text_indexir r r __init__ s
z_Table.__init__c C s dgt | jd }dgt | jd }dgt | jd }dgt | j }||d< x"tdt | jdD ] }|||< qhW x"tdt | jdD ] }|||< qW | jrx|D ] }d|d< qW |S )z
Create z matrix to make heatmap with striped table coloring
:rtype (list[list]) table_matrix: z matrix to make heatmap with striped
table coloring.
r g ?r Nr )r r r9 r, )r< headerZodd_rowZeven_rowr0 r> arrayr r r r) s
z_Table.get_table_matrixc C s t | jdkr | jt | j }nt | jdkrttt | j}| jd |d< x(tdt | jdD ]}| jd ||< qbW xVtdt | jdD ]}| jd ||< qW n,t | jt | jkr| j}ndgt | j }|S )aP
Fill font-color array.
Table text color can vary by row so this extends a single color or
creates an array to set a header color and two alternating colors to
create the striped table pattern.
:rtype (list[list]) all_font_colors: list of font colors for each row
in table.
r r r r z#000000)r r
r listr9 )r< all_font_colorsr> r r r get_table_font_color s z_Table.get_table_font_colorc
C s t | }t | }g }xt| jD ]\}}xt|D ]\}}|dksT| jrd|dk rddt| d nt|}| jr|dkr| jd n|| } |t j
j || j | | j
| j| ddddt| dd d
q6W q$W |S ) z
Generate annotations to fill in table text
:rtype (list) annotations: list of annotations for each cell of the
table.
r r zzx1y1left)colorF) textr: r; ZxrefZyrefalignZxanchorZfontZ showarrow)r( r) rD enumerater r, strr
appendr r' Z
Annotationr: r. r; r+ )
r< r0 rC r# nrowmvalZ format_textZ
font_colorr r r r* s(
& z _Table.make_table_annotationsN)__name__
__module__ __qualname____doc__r? r) rD r* r r r r r( s
r( )NNFr r r r )
__future__r Zplotlyr r Zplotly.graph_objsr Z
get_moduler3 r
r2 objectr( r r r r s