U ARd6 @sddlmZddlmZmZmZmZmZmZm Z m Z m Z ddl Z ddlmZddlmZddlmZddlmZddlmZdd lmZerdd lmZe eee jeeeeeefdfZGd d d Z d*eeeedd ddZ!eeeddddZ"eeeddddZ#eeddddZ$eeeeefddddZ%e deeefdZ&ee&ee&dddZ'd+eeeefeeed!d"d#Z(eeeeefdd$d%d&Z)eeeefed'd(d)Z*dS),)Iterable) TYPE_CHECKINGAnyDictListMappingOptionalTypeVarUnioncastN)ndarray) DataFrame)Styler) type_util)Arrow)gather_metrics)DeltaGeneratorc @sfeZdZeddddeeeeeeddddZed dedd d d Z e dd ddZ dS) ArrowMixin_arrow_dataframeNF)use_container_widthr)datawidthheightrreturncCsZ|j}tt|}t}||_|r,||_|r6||_tjj |_ t ||||j d|S)agDisplay a dataframe as an interactive table. Parameters ---------- data : pandas.DataFrame, pandas.Styler, pyarrow.Table, numpy.ndarray, pyspark.sql.DataFrame, snowflake.snowpark.DataFrame, Iterable, dict, or None The data to display. If 'data' is a pandas.Styler, it will be used to style its underlying DataFrame. width : int or None Desired width of the dataframe element expressed in pixels. If None, the width will be automatically determined. height : int or None Desired height of the dataframe element expressed in pixels. If None, the height will be automatically determined. use_container_width : bool If True, set the dataframe width to the width of the parent container. This takes precedence over the width argument. This argument can only be supplied by keyword. Examples -------- >>> import streamlit as st >>> import pandas as pd >>> import numpy as np >>> >>> df = pd.DataFrame( ... np.random.randn(50, 20), ... columns=('col %d' % i for i in range(20))) ... >>> st._arrow_dataframe(df) >>> st._arrow_dataframe(df, 200, 100) You can also pass a Pandas Styler object to change the style of the rendered DataFrame: >>> df = pd.DataFrame( ... np.random.randn(10, 20), ... columns=('col %d' % i for i in range(20))) ... >>> st._arrow_dataframe(df.style.highlight_max(axis=0)) Zarrow_data_frame) dg_get_delta_path_strstrhash ArrowProtorrrZ EditingModeZ READ_ONLYZ editing_modemarshall_enqueue)selfrrrr delta_path default_uuidprotor%>> import streamlit as st >>> import pandas as pd >>> import numpy as np >>> >>> df = pd.DataFrame( ... np.random.randn(10, 5), ... columns=("col %d" % i for i in range(5))) ... >>> st._arrow_table(df) d)Zmax_unevaluated_rowsZ arrow_table) rZis_snowpark_data_objectZis_typeZ_PYSPARK_DF_TYPE_STRconvert_anything_to_dfrrrrrrr )r!rr"r#r$r%r%r&r'ws   zArrowMixin._arrow_table)rcCs td|S)zGet our DeltaGenerator.r)r )r!r%r%r&rsz ArrowMixin.dg)NNN)N) __name__ __module__ __qualname__rDatarintboolrr'propertyrr%r%r%r&r+s"J*r)r$rr#rcCs\t|r(t|tstdt|||t|tjrBt||_ nt |}t ||_ dS)a[Marshall pandas.DataFrame into an Arrow proto. Parameters ---------- proto : proto.Arrow Output. The protobuf for Streamlit Arrow proto. data : pandas.DataFrame, pandas.Styler, pyarrow.Table, numpy.ndarray, pyspark.sql.DataFrame, snowflake.snowpark.DataFrame, Iterable, dict, or None Something that is or can be converted to a dataframe. default_uuid : Optional[str] If pandas.Styler UUID is not provided, this value will be used. This attribute is optional and only used for pandas.Styler, other elements (e.g. charts) can ignore it. z.Default UUID must be a string for Styler data.N) rZis_pandas_styler isinstancerAssertionErrormarshall_stylerpaTableZpyarrow_table_to_bytesrr)data_frame_to_bytes)r$rr#dfr%r%r&rs    r)r$stylerr#rcCs\t||||tdr(|}n |dd}t||t|||t||j|dS)aoMarshall pandas.Styler into an Arrow proto. Parameters ---------- proto : proto.Arrow Output. The protobuf for Streamlit Arrow proto. styler : pandas.Styler Helps style a DataFrame or Series according to the data with HTML and CSS. default_uuid : str If pandas.Styler uuid is not provided, this value will be used. z1.3.0FN) _marshall_uuidZ_computeris_pandas_version_less_thanZ _translate_marshall_caption_marshall_styles_marshall_display_valuesr)r$r8r#Z pandas_stylesr%r%r&r3s      r3cCs&|jdkr||t|j|j_dS)atMarshall pandas.Styler uuid into an Arrow proto. Parameters ---------- proto : proto.Arrow Output. The protobuf for Streamlit Arrow proto. styler : pandas.Styler Helps style a DataFrame or Series according to the data with HTML and CSS. default_uuid : str If pandas.Styler uuid is not provided, this value will be used. N)uuidZset_uuidrr8)r$r8r#r%r%r&r9s  r9)r$r8rcCs|jdk r|j|j_dS)aMarshall pandas.Styler caption into an Arrow proto. Parameters ---------- proto : proto.Arrow Output. The protobuf for Streamlit Arrow proto. styler : pandas.Styler Helps style a DataFrame or Series according to the data with HTML and CSS. N)captionr8)r$r8r%r%r&r;s r;)r$r8stylesrcCsg}d|krB|d}t|}|D] }td||jdd}||q d|kr||d}t|}|D]}td||j}||q^t|dkrd||j_dS) aRMarshall pandas.Styler styles into an Arrow proto. Parameters ---------- proto : proto.Arrow Output. The protobuf for Streamlit Arrow proto. styler : pandas.Styler Helps style a DataFrame or Series according to the data with HTML and CSS. styles : dict pandas.Styler translated styles. table_styles ) separator cellstyle cell_styler N)_trim_pandas_styles_pandas_style_to_cssr>appendlenjoinr8r@)r$r8r@Z css_rulesrAstyleZrulerDr%r%r&r<s(   r<M)bound)r@rcCsdd|DS)zFilter out empty styles. Every cell will have a class, but the list of props may just be [['', '']]. Parameters ---------- styles : list pandas.Styler translated styles. cSs&g|]}tdd|dDr|qS)css|]}t|VqdS)Nany).0yr%r%r& Hsz1_trim_pandas_styles...propsrO)rQxr%r%r& Hsz'_trim_pandas_styles..r%)r@r%r%r&rG<s rG) style_typerLr>rCrcCsg}|dD]&\}}|d|}||q d|}|dksX|dkrdtdrd|dg} n|d} g} | D]} | ||| qtd | } d |} | d | d }|S) aTConvert pandas.Styler translated style to CSS. Parameters ---------- style_type : str Either "table_styles" or "cell_style". style : dict pandas.Styler translated style. uuid : str pandas.Styler uuid. separator : str A string separator used between table and cell selectors. rTz: z#T_rArEz1.1.0selector selectorsz, z; z { z })striprIrr:rK)rXrLr>rCZ declarationsZ css_propertyZ css_valueZ declarationZtable_selectorZcell_selectorsrZZ cell_selectorrYZdeclaration_blockZrule_setr%r%r&rHKs&     rH)r$r7r@rcCst||}t||j_dS)a0Marshall pandas.Styler display values into an Arrow proto. Parameters ---------- proto : proto.Arrow Output. The protobuf for Streamlit Arrow proto. df : pandas.DataFrame A dataframe with original values. styles : dict pandas.Styler translated styles. N)_use_display_valuesrr6r8Zdisplay_values)r$r7r@new_dfr%r%r&r=s r=)r7r@rc Cs~ddl}|t}|d}d|krz|d}|D]H}|D]>}||d}|r8tt|\} } t|d|j| | f<q8q0|S)zCreate a new pandas.DataFrame where display values are used instead of original ones. Parameters ---------- df : pandas.DataFrame A dataframe with original values. styles : dict pandas.Styler translated styles. rNzrow(\d+)_col(\d+)bodyidZ display_value) reZastypercompilematchmapr.groupsZiat) r7r@r`r]Zcell_selector_regexZrowsrowcellrbrcr%r%r&r\s   r\)N)rW)+collections.abcrtypingrrrrrrr r r Zpyarrowr4Znumpyr Zpandasr Zpandas.io.formats.stylerZ streamlitrZstreamlit.proto.Arrow_pb2rrZstreamlit.runtime.metrics_utilrZstreamlit.delta_generatorrr5rr-rrr3r9r;r<rMrGrHr=r\r%r%r%r&sJ ,        $} #  )  D