U -e>@sUdZddlmZddlmZddlmZmZddlm Z ddl m Z ddl m Z ddlZddlZdd lmZdd lmZmZmZmZmZmZmZmZmZmZmZmZdd lm Z ddl!Z"dd l#m$Z$m%Z%dd l&m'Z'ddl(m)Z)ddl*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0ddl1m2Z2ddl3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>ddl?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKddlLmMZMddlNmOZOmPZPddlQmRZRmSZSmTZTddlUmVZVddlWmXZXddlYmZm[Z\ddl]m^Z^ddl_m`Z`maZambZbmcZcddldmeZeddlfmgZgddlhmiZiddljmkZkmlZlddlmmnZnerNddlompZpmqZqd Zrd!esd"<d#Ztd$Zud!esd%<Gd&d'd'ZvGd(d)d)ZwGd*d+d+ZxGd,d-d-exZyd+d.d/d0Zzd1d.d2d3Z{d1d.d4d5Z|Gd6d7d7Z}Gd8d9d9Z~dd:d;dd?Zedd;dsz2CategoricalFormatter.to_string..rm[]ro)r_rprcrsryjoinappendr])rhr_ fmt_valuesvaluesresultrcririrj to_strings   zCategoricalFormatter.to_string)NTrZT)__name__ __module__ __qualname__rkrsryrriririrjrYs rYc@sxeZdZddddddd dd dd d d d ddZd dddZd dddZddddZddddZd dddZdS)SeriesFormatterNTrZFrMr[ bool | strr\r] str | None int | Noner^) seriesr`raheaderindexrbnamerudtypemax_rowsmin_rowsrdc Cst||_|dk r|nt|_||_||_||_||_||_| |_| |_ |dkrTt d}||_ | |_ t |_|dS)Ndisplay.float_format)rrr`rrbrrarrrrrurget_adjustmentadj _chk_truncate) rhrr`rarrrbrrurrrririrjrk szSeriesFormatter.__init__rlcCs||j}|j}|o t|j|k}|j}|rtt|}|rDt||}|dkr`|}|jd|}n*|d}t|jd||j| df}||_ nd|_ ||_ ||_ dS)N) rrrprrintminilocrH tr_row_num tr_seriesis_truncated_vertically)rhrrrrrow_numririrjr)s$  "zSeriesFormatter._chk_truncatecCsF|jj}d}t|jjdddk rLt|jjtttfs8t|d|jjj 7}|jdk r|dk r|rj|d7}t j |dd}|d|7}|j d ks|j d kr|j r|r|d7}|d t|j7}|jdk r|jdk rt|jjd d}|r|r|d7}|d t |7}t|jjr>|jj}|r6|d7}||7}t|S)NrffreqzFreq: Frm  ro) escape_charszName: Ttruncaternrzdtype: ro)rrgetattrr isinstancerFrDrGAssertionErrorZfreqstrrK pprint_thingrarrprrr-_valuesrqr])rhrrcZ series_nameZ dtype_namerrririrjrsDsB  zSeriesFormatter._get_footerztuple[list[str], bool]cCsR|jj}t|tr4tdd|jD}|jdd}n|jdk }|jdd}||fS)Ncss|] }|VqdSNrir|rririrj qsz7SeriesFormatter._get_formatted_index..T)names)r)rrrrCanyrformatr)rhr have_header fmt_indexririrj_get_formatted_indexms   z$SeriesFormatter._get_formatted_indexrtcCst|jjd|j|j|jdS)N)rurb leading_space)rvrrrurbrrxririrjryxsz%SeriesFormatter._get_formatted_valuesc Cs.|j}|}t|dkr2t|jjd|dS|\}}|}|jrd}|j }t t |}|j ||d}|dkrd} nd} |j j | g|dd d} |||| ||dd |jr|j jd |dd|g} n|j d|} |jr|r|dd | } |r | d |7} td | S) Nrz([], )r...z..rQmoderfro)r)rrsrptyperrrryrrrrrrRinsertradjoinrr]r) rhrrcrrrZ n_header_rowsrwidthZdot_strrririrjrs2    zSeriesFormatter.to_string) NTTTrZFNTNN) rrrrkrrsrryrriririrjrs & )  rc@sPeZdZddddZddddd Zdd ddd d ddZdddddZdS)TextAdjustmentr^rlcCstd|_dS)Nzdisplay.encoding)rencodingrxririrjrkszTextAdjustment.__init__r]rtextrdcCst|SrrprhrririrjrpszTextAdjustment.lenrPr rttextsmax_lenrrdcCstj|||dS)Nr)rKrRrhrrrririrjrRszTextAdjustment.justify)spacerdcOs tj|f||j|jd|S)N)ZstrlenZjustfunc)rKrrprR)rhrlistskwargsririrjrszTextAdjustment.adjoinN)rP)rrrrkrprRrriririrjrsrcsHeZdZddfdd Zddddd Zdd ddd d ddZZS)EastAsianTextAdjustmentr^rlcs6ttdrd|_nd|_dddddd|_dS)Nz!display.unicode.ambiguous_as_widerr)ZNaNWFH)superrkrambiguous_width_EAW_MAPrx __class__rirjrks  z EastAsianTextAdjustment.__init__r]rrcs(t|tst|Stfdd|DS)zN Calculate display width considering unicode East Asian Width c3s"|]}jt|jVqdSr)rgetrr)r|crxrirjrsz.EastAsianTextAdjustment.len..)rr]rpsumrrirxrjrps   zEastAsianTextAdjustment.lenrPz Iterable[str]rtrcsXfdd|dkr(fdd|DS|dkrBfdd|DSfdd|DSdS) Ncs|t|Srr)t)rrhrirj_get_padsz1EastAsianTextAdjustment.justify.._get_padrOcsg|]}||qSri)ljustr|xrrirjr~sz3EastAsianTextAdjustment.justify..rQcsg|]}||qSri)rQrrrirjr~scsg|]}||qSri)rjustrrrirjr~srirri)rrrhrjrRs zEastAsianTextAdjustment.justify)rP)rrrrkrprR __classcell__ririrrjrs  rrlcCstd}|rtStSdS)Nz display.unicode.east_asian_width)rrr)Zuse_east_asian_widthririrjrsrzdict[str, Any]cCsLddlm}tdr"|\}}nd}tdtdtdtdtd |d S) aGet the parameters used to repr(dataFrame) calls using DataFrame.to_string. Supplying these parameters to DataFrame.to_string is equivalent to calling ``repr(DataFrame)``. This is useful if you want to adjust the repr output. .. versionadded:: 1.4.0 Example ------- >>> import pandas as pd >>> >>> df = pd.DataFrame([[1, 2], [3, 4]]) >>> repr_params = pd.io.formats.format.get_dataframe_repr_params() >>> repr(df) == df.to_string(**repr_params) True r)consolezdisplay.expand_frame_reprNdisplay.max_rowsdisplay.min_rowszdisplay.max_columnsdisplay.max_colwidthdisplay.show_dimensions)rrmax_colsZ max_colwidthshow_dimensions line_width)pandas.io.formatsrrZget_console_size)rr_ririrjget_dataframe_repr_paramss rcCsNt\}}tddkr|ntd}tddkr2|ntd}dd||tddS)aGet the parameters used to repr(Series) calls using Series.to_string. Supplying these parameters to Series.to_string is equivalent to calling ``repr(series)``. This is useful if you want to adjust the series repr output. .. versionadded:: 1.4.0 Example ------- >>> import pandas as pd >>> >>> ser = pd.Series([1, 2, 3, 4]) >>> repr_params = pd.io.formats.format.get_series_repr_params() >>> repr(ser) == ser.to_string(**repr_params) True rrrTr)rrrrra)r r)rheightrrririrjget_series_repr_paramss   rc@sXeZdZdZerendZeee7Zdjdd d d d d ddddd ddddd d d ddddZddddZed dddZ ed dddZ ed dd d!Z ed dd"d#Z ed dd$d%Z ed dd&d'Zed dd(d)Zed dd*d+Zed dd,d-Zed.dd/d0Zdd d1d2d3Zdd4d5d6d7Zdd d8d9d:Zd d;d<d=d>Zd d?d@dAdBZdddCdDZdddEdFZdddGdHdIZd ddJdKZd ddLdMZd ddNdOZd.ddPdQZdddRdSZdddTdUZ dddVdWZ!dddXdYZ"d.dZd[d\d]Z#d^d_d[d`daZ$dddbdcddZ%ddZdbdedfZ&dgddhdiZ'dS)kDataFrameFormatterz;Class for processing dataframe formatting options and data.rfNTrZF.rLSequence[Hashable] | NonezColspaceArgType | Nonezbool | Sequence[str]r\r]zFormattersType | NonerFloatFormatType | None bool | Nonerrr^)framecolumns col_spacerrrb formattersrRrusparsify index_namesrrrrdecimal bold_rowsescaperdcCs||_|||_|||_||_||_||_|||_ | ||_ | |_ | | |_| |_||_||_||_| |_| |_||_||_||_||_|j|_|t|_dSr)r_initialize_columnsr_initialize_colspacerrrrb_initialize_formattersr_initialize_justifyrRru_initialize_sparsifyrshow_index_namesrrrrrrr_calc_max_cols_fittedmax_cols_fitted_calc_max_rows_fittedmax_rows_fittedtr_framerrr)rhrrrrrrbrrRrurrrrrrrrrririrjrk4s.       zDataFrameFormatter.__init__zlist[list[str]]rlcCs*|}|jr&||j}|d||S)zP Render a DataFrame to a list of columns (as lists of strings). r)_get_strcols_without_indexrrrr)rhstrcolsZ str_indexririrj get_strcolscs   zDataFrameFormatter.get_strcolscCs|jdkp|jdko|jS)NTr)r is_truncatedrxririrjshould_show_dimensionsos z)DataFrameFormatter.should_show_dimensionscCst|jp |jSr)r\is_truncated_horizontallyrrxririrjr uszDataFrameFormatter.is_truncatedcCst|jot|j|jkSr)r\rrprrxririrjr ysz,DataFrameFormatter.is_truncated_horizontallycCst|jot|j|jkSr)r\rrprrxririrjr}sz*DataFrameFormatter.is_truncated_verticallycCs dt|jdt|jjdS)Nz [z rows x z columns])rprrrxririrjdimensions_infosz"DataFrameFormatter.dimensions_infocCs t|jjSr) _has_namesrrrxririrjhas_index_namessz"DataFrameFormatter.has_index_namescCs t|jjSr)rrrrxririrjhas_column_namessz#DataFrameFormatter.has_column_namescCst|j|j|jfSr)allrrrrxririrjshow_row_idx_namessz%DataFrameFormatter.show_row_idx_namescCst|j|j|jfSr)rrrrrxririrjshow_col_idx_namessz%DataFrameFormatter.show_col_idx_namesrcCst|jpt|jt|jSr)rrrprrxririrjmax_rows_displayedsz%DataFrameFormatter.max_rows_displayed)rrdcCs|dkrtdS|S)Nzdisplay.multi_sparser)rhrririrjrsz'DataFrameFormatter._initialize_sparsifyr))rrdcCsT|dkr iSt|jjt|ks*t|tr.|Stdt|dt|jjddS)NzFormatters length(+) should match DataFrame number of columns(r)rprrrdict ValueError)rhrririrjrsz)DataFrameFormatter._initialize_formatters)rRrdcCs|dkrtdS|SdS)Nzdisplay.colheader_justifyr)rhrRririrjrsz&DataFrameFormatter._initialize_justifyrB)rrdcCs2|dk r&ttt|}|j||_|S|jjSdSr)rErr#rr)rhrcolsririrjrs  z&DataFrameFormatter._initialize_columnsr%)rrdcsdkri}ntttfrBdi}|fdd|jjDnttrD]&}||jjkrT|dkrTtd|qT}nHt |jjt krtdt dt |jjdt t |jj}|S)Nrfcsi|] }|qSriri)r|columnrrirj sz;DataFrameFormatter._initialize_colspace..z,Col_space is defined for an unknown column: zCol_space length(rr) rrr]updaterrrkeysrrprzip)rhrrrrirrjrs$  z'DataFrameFormatter._initialize_colspacecCs0|s|jSt\}}||r&|S|jSdS)z%Number of columns fitting the screen.N)_is_in_terminalrr _is_screen_narrow)rhrrririrjrs   z(DataFrameFormatter._calc_max_cols_fittedcCsP|r@t\}}|jdkr(||S||r8|}qF|j}n|j}||S)z,Number of rows with data fitting the screen.r)r!r r_get_number_of_auxillary_rows_is_screen_short_adjust_max_rows)rhrrrririrjrs    z(DataFrameFormatter._calc_max_rows_fitted)rrdcCs(|r$t|j|kr$|jr$t|j|}|S)zAdjust max_rows using display logic. See description here: https://pandas.pydata.org/docs/dev/user_guide/options.html#frequently-used-options GH #37359 )rprrr)rhrririrjr%s z#DataFrameFormatter._adjust_max_rowscCst|jdkp|jdkS)z/Check if the output is to be shown in terminal.r)r\rrrxririrjr!sz"DataFrameFormatter._is_in_terminalcCst|jdkot|jj|kSNr)r\rrprr)rh max_widthririrjr"sz$DataFrameFormatter._is_screen_narrowcCst|jdkot|j|kSr&)r\rrpr)rh max_heightririrjr$sz#DataFrameFormatter._is_screen_shortcCs:d}d}||}|jr(|t|j7}|jr6|d7}|S)z?Get number of rows occupied by prompt, dots and dimension info.r)rrpr splitlinesr)rhZdot_rowZ prompt_rowZnum_rowsririrjr# sz0DataFrameFormatter._get_number_of_auxillary_rowscCs |jr||jr|dS)zY Check whether the frame should be truncated. If so, slice the frame up. N)r _truncate_horizontallyr_truncate_verticallyrxririrjrszDataFrameFormatter.truncatecCs|jdk st|jd}|dkr|jjddd|f}|jjdd| df}t||fdd|_t|jttfr|jd||j| d|_n&t t |j }|jjddd|f|_||_ dS)zRemove columns, which are not to be displayed and adjust formatters. Attributes affected: - tr_frame - formatters - tr_col_num NrrZaxis) rrrrrHrrlisttuplerrrZ tr_col_num)rhZcol_numrOrPririrjr*!s   z)DataFrameFormatter._truncate_horizontallycCs|jdk st|jd}|dkrb|jjd|ddf}|jj| dddf}t||f|_n&tt|j}|jjd|ddf|_||_dS)zRemove rows, which are not to be displayed. Attributes affected: - tr_frame - tr_row_num Nrr) rrrrrHrrrr)rhrheadtailririrjr+;s  z'DataFrameFormatter._truncate_verticallyc sg}tjsbjsbtjD]>\}}|}t|jtj |dj d}| |q|Stjrt t tj_tjtjkrtdtjdtjdddjD}n j}jr|D]}| dqtjD]\}}||}ttj |dffd d |D}|}t|j|j d }ttfd d |D|} j j|| jd }| ||q|S)Nr)stringsrRminimumrzWriting z cols but got z aliasescSsg|] }|gqSriri)r|labelririrjr~dszADataFrameFormatter._get_strcols_without_index..rfc3s|]}j|VqdSrrrprrxrirjrosz@DataFrameFormatter._get_strcols_without_index..)r2rc3s|]}j|VqdSrr4rrxrirjrvsr)r5r enumerater format_col_make_fixed_widthrRrrrrrrrr]rprr_get_formatted_column_labelsrmax) rhr r}rr str_columnsrZcheaderZheader_colwidthrrirxrjrMsR      z-DataFrameFormatter._get_strcols_without_indexrt)r}rdc CsL|j}||}t|jdd|fj||j|j|j|j ||j |j dS)N)rurbrrr) r_get_formatterrvrrrurbrrrrr)rhr}r formatterririrjr6|s zDataFrameFormatter.format_col str | intCallable | NonecCs`t|jttfr2t|r,tt|}|j|SdSn*t|rN||jkrN|j|}|j|dSdSr) rrr-r.r3rrrr)rhr}ririrjr;s   z!DataFrameFormatter._get_formatter)rrdcsddlm}|j}t|tr|jddd}tt|}jj j }t dd|j Dt t|tt|fddttfd d|D}jrt|r||}d d t|D}n<|}jj }t t|tt|fd d t|D}|S) Nr)sparsify_labelsF)rrcss|] }|jVqdSr)Z is_floating)r|levelririrjrszBDataFrameFormatter._get_formatted_column_labels..cs"|jkr|rsd|S|S)N )r)ry) need_leadsprestrict_formattingrhrirj space_formatszEDataFrameFormatter._get_formatted_column_labels..space_formatc3s"|]fddDVqdS)csg|]}|qSriri)r|rB)rErrirjr~szMDataFrameFormatter._get_formatted_column_labels...Nri)r|)rErrjrscSsg|] }t|qSri)r-rririrjr~szCDataFrameFormatter._get_formatted_column_labels..cs0g|](\}}|s&|r&d|n|gqS)rA)r;)r|r}r)rCrhrirjr~s)Zpandas.core.indexes.multir?rrrCrr-r rdtypesrrlevelsrmapr6rrpr5)rhrr?rZ fmt_columnsrGr:ri)rCrDrhrErjr8s,      z/DataFrameFormatter._get_formatted_column_labelscsddjD|j}|j}d}t|trL|jjdj |d}n|jj |dg}fdd|D}j j d | d }j rd dD}n d g|j}jr||S|SdS)NcSsi|]\}}|tt|qSri)rr)r|kvririrjrsz;DataFrameFormatter._get_formatted_index.. __index__F)rrrr<)rr<c s.g|]&}ttt|dddjdqS)rOrfr)rRr2r)r.r7r-rrrrrhrirjr~s z;DataFrameFormatter._get_formatted_index..rrocSsg|] }t|qSrir]rririrjr~srf)r)ritemsrrr;rrCrrrrrsplitr_get_column_name_listZnlevelsr)rhrrrfmtrZadjoinedZ col_headerrirMrjrs,     z'DataFrameFormatter._get_formatted_indexzlist[Hashable]cCsLg}|jj}t|tr.|dd|jDn||jdkr@dn|j|S)Ncss|]}|dkrdn|VqdSrerirririrjrsz;DataFrameFormatter._get_column_name_list..rf)rrrrCextendrrr)rhrrririrjrQs  z(DataFrameFormatter._get_column_name_list)NNTTrZNNNNTNNNFrFT)(rrr__doc__rNrXrkr propertyr r r rrrrrrrrrrrrrrr%r!r"r$r#rr*r+rr6r;r8rrQriririrjr.s~  4/     /  )(rc@seZdZdZdddddZd(d d d d d d d d d d d d ddZd)d d dd dd d d dddZd*d d dd dddZd+dd d d!d"d d#dd d dd d d d d$d d%d&d'ZdS),DataFrameRendereraJClass for creating dataframe output in multiple formats. Called in pandas.core.generic.NDFrame: - to_csv - to_latex Called in pandas.core.frame.DataFrame: - to_html - to_string Parameters ---------- fmt : DataFrameFormatter Formatter with the formatting options. rr^)rRrdcCs ||_dSr)rR)rhrRririrjrkszDataFrameRenderer.__init__NF"FilePath | WriteBuffer[str] | Nonerr\zstr | tuple[str, str] | None) r` column_format longtabler multicolumnmulticolumn_formatmultirowcaptionr3positionrdc  Cs>ddlm} | |j||||||| | d } | } t| ||dS)zU Render a DataFrame to a LaTeX tabular/longtable environment output. r)LatexFormatter)rYrXrZr[r\r]r3r^r`r)Zpandas.io.formats.latexr_rRrsave_to_buffer)rhr`rXrYrrZr[r\r]r3r^r_Zlatex_formatterstringririrjto_latexs  zDataFrameRenderer.to_latexzstr | list | tuple | Nonezint | bool | None)r`rclassesnotebookbordertable_id render_linksrdc CsFddlm}m} |r| n|} | |j||||d} | } t| ||dS)uJ Render a DataFrame to a html table. Parameters ---------- buf : str, path object, file-like object, or None, default None String, path object (implementing ``os.PathLike[str]``), or file-like object implementing a string ``write()`` function. If None, the result is returned as a string. encoding : str, default “utf-8” Set character encoding. classes : str or list-like classes to include in the `class` attribute of the opening ```` tag, in addition to the default "dataframe". notebook : {True, False}, optional, default False Whether the generated HTML is for IPython Notebook. border : int A ``border=border`` attribute is included in the opening ``
`` tag. Default ``pd.options.display.html.border``. table_id : str, optional A css id is included in the opening `
` tag if specified. render_links : bool, default False Convert URLs to HTML links. r) HTMLFormatterNotebookFormatter)rdrfrgrhr`)Zpandas.io.formats.htmlrirjrRrra) rhr`rrdrerfrgrhrirjZKlassZhtml_formatterrbririrjto_html$s" zDataFrameRenderer.to_htmlr)r`rrrdcCs0ddlm}||j|d}|}t|||dS)u% Render a DataFrame to a console-friendly tabular output. Parameters ---------- buf : str, path object, file-like object, or None, default None String, path object (implementing ``os.PathLike[str]``), or file-like object implementing a string ``write()`` function. If None, the result is returned as a string. encoding: str, default “utf-8” Set character encoding. line_width : int, optional Width to wrap a line in characters. r)StringFormatter)rr`)Zpandas.io.formats.stringrlrRrra)rhr`rrrlZstring_formatterrbririrjrWs zDataFrameRenderer.to_string,winfer"Tstrictz7FilePath | WriteBuffer[bytes] | WriteBuffer[str] | Noner]rzIndexLabel | Noner&r+) path_or_bufrsepr index_labelr compressionrg quotecharlineterminator chunksize date_format doublequote escapecharerrorsstorage_optionsrdcCsddlm}|dkr d}t}nd}||| ||||||||| | | | |||jd}||r~t|tsjt|}||SdS)z; Render dataframe as comma-separated file. r) CSVFormatterNTF)rrrwrsrr|rurgrrtrrxrvryrzr{r}r<) Zpandas.io.formats.csvsr~rrRsaverrgetvalueclose)rhrrrrsrrtrrurgrvrwrxryrzr{r|r}r~Zcreated_bufferZ csv_formattercontentririrjto_csvqs> zDataFrameRenderer.to_csv) NNFNFNFNNN)NNNFNNF)NNN)NNrmNNrnroNrpNNNTNrqN) rrrrTrkrcrkrrriririrjrVsT$"5rVr]rWr)rbr`rrdc CsLt||d6}|||dkr4|W5QRSW5QRdSQRXdS)zQ Perform serialization. Write to buf or return as string if buf is None. )rN) get_bufferwriter)rbr`rfririrjras  razIGenerator[WriteBuffer[str], None, None] | Generator[StringIO, None, None])r`rrdc cs|dk rt|}nt}|dkr&d}nt|ts8tdt|drJ|VnBt|trtt|t|d|dd }|VW5QRXntddS) z Context manager to open, yield and close buffer for filenames or Path-like objects, otherwise yield buf unchanged. Nzutf-8z1buf is not a file name and encoding is specified.rrnrf)rnewlinez1buf is not a file name and it has no write method) rJrrr]rhasattrrIopen TypeError)r`rrririrjrs     rrZrPrTr r>rrzstr | int | Nonerrt) rr<rurbdigitsrrRrrrgfallback_formatterrdc Cst|jrt} n`t|jtr"t} nNt|jr2t} n>t|jrBt } n.t |jsVt |jr\t } nt |jrlt} nt} |dkr|d}|dkrtd}|dkrtd}| |||||||||| | d } | S)a{ Format an array for printing. Parameters ---------- values formatter float_format na_rep digits space justify decimal leading_space : bool, optional, default True Whether the array should be formatted with a leading space. When an array as a column of a Series or DataFrame, we do want the leading space to pad between columns. When formatting an Index subclass (e.g. IntervalIndex._format_native_types), we don't want the leading space since it should be left-aligned. fallback_formatter Returns ------- List[str] N rdisplay.precision) rrbrur<rrRrrrgr)r/rDatetime64Formatterrr9Datetime64TZFormatterr8Timedelta64Formatterr0ExtensionArrayFormatterr2r.FloatArrayFormatterr4IntArrayFormatterGenericArrayFormatterr get_result) rr<rurbrrrRrrrgrZ fmt_klassZfmt_objririrjrvsB)     rvc@sPeZdZddd d d d d d d dddd dd ddZddddZddddZdS)rNrZrrPrTr rr>r]r=rrr\rr^) rrr<rbrrurRrrg fixed_widthrrrdc CsL||_||_||_||_||_||_||_||_| |_| |_ | |_ | |_ dSr) rrrbrr<rurRrrgrrr) rhrrr<rbrrurRrrgrrrririrjrk6szGenericArrayFormatter.__init__rtrlcCs|}t||jSr)_format_stringsr7rR)rhrririrjrRsz GenericArrayFormatter.get_resultc sjdkr0td}|dkr6tdfdd}nj}jdk rHjn6jdk rZjn$jdk oljtk}ttjd|dfdd}t j d d }t |t j std t|t}|t jt|ttd t|jd @}j}|dkr|}g}t|D]\} } || r jdk r<|r<|d|| nD|| rV||| n*|dkrfd} nd} || j|| dq|S)Nrrcst|dddS)Nz .dr)_trim_zeros_single_floatrF) precisionrirj[sz7GenericArrayFormatter._format_strings..r)r quote_stringsc sjdk rvt|rvt|rvz<|dkr*WdS|tkr._formatTZ extract_numpyz.base_formattercst|rt|SjSr)r;r]rbrrxrirjrsrcs|djdS)Nrr)replacerr)rrhrirjdecimal_formattersz?FloatArrayFormatter._value_formatter..decimal_formattercs0t|r&t|kr|SdSnjSdS)Ng)r;absrbr)rrhrrirjr<s   z7FloatArrayFormatter._value_formatter..formatter)rur)rhrurr<ri)rrrurhrrj_value_formatters   z$FloatArrayFormatter._value_formatter np.ndarrayc spddddddjdk r.jjjSjr>tdndfd d }jdkrjrjd krrd }nd }t|jj d}qj}n fdd}||}js|St |dkrt dd|D}|j dk}nd}t j dd:t j}|dk}|dj k|dk@} W5QRX| s>|rl|rljd krPd}nd}t|jj d}||}|S)z Returns the float values converted into strings using the parameters given at initialisation, as a numpy array r"r r])rr<rbcs<t|}tfddt||D|j}|S)Ncs g|]\}}|s|nqSriri)r|valmr<rbrirjr~szWFloatArrayFormatter.get_result_as_array..format_with_na_rep..)r:rarrayr ravelZreshaper)rr<rbmask formattedrirrjformat_with_na_reps zCFloatArrayFormatter.get_result_as_array..format_with_na_repNzdisplay.chop_thresholdcsx|}jdkr"dj}nj}j}t|}|||}jrt|rZt|j}n t|j}t j |ddS|S)NrOrAobjectr) rrRrbrr.r_trim_zeros_complexr_trim_zeros_floatrasarray)rur<rbrZ is_complexrrrhrrirjformat_values_withs     zCFloatArrayFormatter.get_result_as_array..format_values_withTz{value: .{digits:d}f}z{value:.{digits:d}f})rcs j|Sr)rurrxrirjr(z9FloatArrayFormatter.get_result_as_array..rcss|]}t|VqdSrrrririrjr4sz:FloatArrayFormatter.get_result_as_array..Fignoreinvalidg.A z{value: .{digits:d}e}z{value:.{digits:d}e})r<rrbrrrurrrrrpr9rerrstaterr) rhrZfmt_strruZformatted_valuesmaxlenZtoo_longZabs_valsZhas_large_valuesZhas_small_valuesrirrjget_result_as_arraysD         z'FloatArrayFormatter.get_result_as_arrayrtcCs t|Sr)r-rrxririrjrMsz#FloatArrayFormatter._format_strings)NN)rrrrkrrrrririrrjrs 8crc@seZdZddddZdS)rrtrlcs>|jdkrdd}ndd}|jp$|fdd|jD}|S)NFcSs|dj|dS)NrrFrrFririrjrTrz3IntArrayFormatter._format_strings..cSs|dj|dS)Nz drFrrFririrjrVrcsg|] }|qSririrr<rirjr~Xsz5IntArrayFormatter._format_strings..)rr<r)rhZ formatter_strrrirrjrRs    z!IntArrayFormatter._format_stringsNrrrrriririrjrQsrcs8eZdZd dddddfdd Zd d d d ZZS)rrNz3np.ndarray | Series | DatetimeIndex | DatetimeArrayr]r^)rnat_repryrdc s tj|f|||_||_dSr)rrkrry)rhrrryrrrirjrk]szDatetime64Formatter.__init__rtrlcsZj}t|tst|}jdk r>tjr>fdd|DS|jjjjd}| S)z&we by definition have DO NOT have a TZNcsg|]}|qSrirrrxrirjr~psz7Datetime64Formatter._format_strings..)rbry) rrrFr<r_dataZ_format_native_typesrrytolist)rhrrrirxrjrhs z#Datetime64Formatter._format_strings)rNrrrrkrrririrrjr\s rc@seZdZddddZdS)rrtrlc Cs|t|jdd}|j}d}|dkr,|jdd}t|tr@|}n t|}t |||j |j |j |j |j|j|j|j|d }|S)NTr)Zboxed) rurbrrrRrrrgr)rArr< _formatterrr<rwrrrvrurbrrrRrrrg)rhrr<rrrririrjrys,     z'ExtensionArrayFormatter._format_stringsNrriririrjrxsrznp.ndarray | Sequence[float]) percentilesrdc CsXt|}tjdd2t|r>> format_percentiles([0.01999, 0.02001, 0.5, 0.666666, 0.9999]) ['1.999%', '2.001%', '50%', '66.667%', '99.99%'] No element is rounded to 0% or 100% (unless already equal to it). Duplicates are allowed: >>> format_percentiles([0, 0.5, 0.02001, 0.5, 0.666666, 0.9999]) ['0%', '50%', '2.0%', '50%', '66.67%', '99.99%'] rrrrz/percentiles should all be in the interval [0,1]dcSsg|] }|dqS%rir{ririrjr~sz&format_percentiles..N)to_beginto_endrcSsg|] }|dqSrrir{ririrjr~s)rrrr6rrroundastyperiscloser]uniquefloorlog10rZediff1dr9Z empty_liker)rZpercentiles_round_typeZint_idxoutZ unique_pctsrrprecririrjformat_percentiless8"        rz2np.ndarray | DatetimeArray | Index | DatetimeIndexr\rrdcCs~t|ts|}t|ttfs(t|}|jdk r6dS|j}|tk}t|j }t |}t |||dk dk}|rzdSdS)NFrT)rrBrr=rFtzZasi8rrrr r logical_andr)r values_intconsider_valuesZresoZppd even_daysririrj is_dates_onlys   rrzNaTType | Timestamp)rrrdcCs|tkr |St|Sr)rr])rrririrj_format_datetime64sr)rrryrdcCs&t|tr|S|r||S|jSdSr)rr!strftimeZ _date_repr)rrryririrj_format_datetime64_dateonlys   rr )is_dates_only_rryrdcs"|rfddSfddSdS)z]Return a formatter callable taking a datetime64 as input and providing a string as outputcst|dS)N)rry)rrFryrrirjrsz'get_format_datetime64..cs t|dS)Nr)rrFrrirjrrNri)rrryrirrjget_format_datetime64srz*np.ndarray | DatetimeArray | DatetimeIndex)rryrdcCs6t|tjr|jdkr|}t|}|r2|p0dS|S)z6given values and a date_format, return a string formatrz%Y-%m-%d)rrrndimrr)rryidoririrj!get_format_datetime64_from_valuess rc@seZdZddddZdS)rrtrlcs>|jt}t|}|jp&t||jdfdd|D}|S)zwe by definition have a TZ)rycsg|] }|qSririrrrirjr~4sz9Datetime64TZFormatter._format_strings..)rrrrr<rry)rhrrrrirrjr-s z%Datetime64TZFormatter._format_stringsNrriririrjr,srcs8eZdZddddddfdd Zd d d d ZZS)rrFznp.ndarray | TimedeltaIndexr]r\r^rrboxrdc s tj|f|||_||_dSr)rrkrr)rhrrrrrrirjrk:szTimedelta64Formatter.__init__rtrlcs.|jpt|j|j|jdfdd|jDS)N)rrcsg|] }|qSririrrrirjr~Isz8Timedelta64Formatter._format_strings..)r<get_format_timedelta64rrrrxrirrjrEs z$Timedelta64Formatter._format_strings)rFrririrrjr9s rFz,np.ndarray | TimedeltaIndex | TimedeltaArrayz str | floatrc s^|tj}|tk}d}||dk}t||}|dk}|rFdndfdd} | S)z Return a formatter function for a range of timedeltas. These will all have the same format argument If box, then show the return in quotes l"R:rNlongcsN|dkst|rt|rSt|ts.t|}|jd}rJd|d}|S)Nr')r7r:rrZ _repr_base)rrrrrrirjrks   z*get_format_timedelta64.._formatter)viewrZint64rrr) rrrrrZ one_day_nanosZ not_midnightZbothrrrirrjrLs     rzTextAdjustment | None)r1rRr2rrdcst|dks|dkr|S|dkr(tn|tfdd|D|dk rTt|tddk rpkrpdddfdd fd d |D}j||d }|S) Nrrc3s|]}|VqdSrrr) adjustmentrirjrsz$_make_fixed_width..rr])rrdcs6dk r2dk|k@r2|ddd}|S)NrrrrF)rconf_maxrrirjjustsz_make_fixed_width..justcsg|] }|qSririr)rrirjr~sz%_make_fixed_width..r)rprr9rrR)r1rRr2rrri)rrrrrjr7{s r7r) str_complexesrrdcsDfdd|D}dd|D}t|fddt||D}|S)z Separates the real and imaginary parts from the complex number, and executes the _trim_zeros_float method on each of those. c s$g|]}dttd|qS)rfz([j+-]))rrrerPr)rrirjr~sz'_trim_zeros_complex..cSsg|] }t|qSrir)r|sririrjr~scsg|]\}}|d|ddd |dd||ddd |dd ||dd d|dd|dqS)Nrr0rri)r|rrJ) max_lengthrirjr~s")r9r )rrtrimmedlengthspaddedri)rrrjrs   r) str_floatrdcCs |d}|dr|d7}|S)zX Trims trailing zeros after a decimal point, leaving just one if necessary. rr)rstripendswith)rririrjrs  rnp.ndarray | list[str]) str_floatsrrdcsp|}tddddfdd dddfd d }||rXfd d |D}q<fd d |D}|S)z Trims the maximum number of trailing zeros equally from all numbers containing decimals, leaving just one if necessary. z^\s*[\+-]?[0-9]+\z[0-9]*$r\rlcst|dk Sr)rmatchrF) number_regexrirjis_number_with_decimalsz1_trim_zeros_float..is_number_with_decimalr rcs0fdd|D}t|dko.tdd|DS)z Determine if an array of strings should be trimmed. Returns True if all numbers containing decimals (defined by the above regular expression) within the array end in a zero, otherwise returns False. csg|]}|r|qSririrrrirjr~sz:_trim_zeros_float..should_trim..rcss|]}|dVqdS)rNr rririrjrsz9_trim_zeros_float..should_trim..)rpr)rnumbersrrirj should_trimsz&_trim_zeros_float..should_trimcs$g|]}|r|ddn|qS)Nrrirrrirjr~sz%_trim_zeros_float..cs*g|]"}|r"|r"|dn|qS)rrr)rrrirjr~s)rcompile)r rrrrri)rrr rjrs  rrB)rrdcCs$t|trtj|jS|jdk SdSr)rrCcomZ any_not_nonerr)rririrjrs  rc@s\eZdZdZdddddddd d d d d ddddddZd!ddddddZddddd ZdS)" EngFormatterzl Formats float values according to engineering format. Based on matplotlib.ticker.EngFormatter rBzarpnurrfrJMGTPEZY)iiiiiiirrr rNFrr\r^accuracyuse_eng_prefixrdcCs||_||_dSrr)r*)rhr)r*ririrjrk szEngFormatter.__init__floatr])numrdc Cstt|}t|rdSt|r(dSd}|dkr>d}| }|dkrfttt|dd}ntd}|t |j }| t|j }t|}|j r|j |}n$|dkrd| d}n d |d}||d |}|j d krd }nd |j dd}|j||d} | S)a Formats a number in engineering notation, appending a letter representing the power of 1000 of the original number. Some examples: >>> format_eng = EngFormatter(accuracy=0, use_eng_prefix=True) >>> format_eng(0) ' 0' >>> format_eng = EngFormatter(accuracy=1, use_eng_prefix=True) >>> format_eng(1_000_000) ' 1.0M' >>> format_eng = EngFormatter(accuracy=2, use_eng_prefix=False) >>> format_eng("-1e-6") '-1.00E-06' @param num: the value to represent @type num: either a numeric value or a string that can be converted to a numeric value (as per decimal.Decimal constructor) @return: engineering formatted string rZinfrrrrzE-02dzE+rNz{mant: g}{prefix}z{mant: .rz f}{prefix})mantprefix)rr]is_nan is_infinitermathrrrr9 ENG_PREFIXESrr*r)r) rhr-ZdnumsignZpow10Z int_pow10r1r0 format_strrririrj__call__s4       zEngFormatter.__call__)NF)rrrrTr5rkr8riririrjrs.rrrr^r(cCstdt||dS)a Format float representation in DataFrame with SI notation. Parameters ---------- accuracy : int, default 3 Number of decimal digits after the floating point. use_eng_prefix : bool, default False Whether to represent a value with SI prefixes. Returns ------- None Examples -------- >>> df = pd.DataFrame([1e-9, 1e-3, 1, 1e3, 1e6]) >>> df 0 0 1.000000e-09 1 1.000000e-03 2 1.000000e+00 3 1.000000e+03 4 1.000000e+06 >>> pd.set_eng_float_format(accuracy=1) >>> df 0 0 1.0E-09 1 1.0E-03 2 1.0E+00 3 1.0E+03 4 1.0E+06 >>> pd.set_eng_float_format(use_eng_prefix=True) >>> df 0 0 1.000n 1 1.000m 2 1.000 3 1.000k 4 1.000M >>> pd.set_eng_float_format(accuracy=1, use_eng_prefix=True) >>> df 0 0 1.0n 1 1.0m 2 1.0 3 1.0k 4 1.0M >>> pd.set_option("display.float_format", None) # unset option rN)rrr+ririrjset_eng_float_formatNs7r9rfzbool | object | strzlist[dict[int, int]])rHsentinelrdc Cst|dkrgSdgt|d}g}|D]b}d}i}t|D]2\}}||rX||krXq>d||<||||<|}q>t||||<||q*|S)a For each index in each level the function returns lengths of indexes. Parameters ---------- levels : list of lists List of values on for level. sentinel : string, optional Value which states that no new index starts on there. Returns ------- Returns list of maps. For each level returns map of indexes (key is index in row and value is length of index). rTF)rpr5r) rHr:controlrr@ last_indexrr}keyririrjget_level_lengthss    r>zWriteBuffer[str])r`linesrdcCs4tdd|Dr dd|D}|d|dS)z Appends lines to a buffer. Parameters ---------- buf The buffer to write to lines The lines to append. css|]}t|tVqdSr)rr]rririrjrsz#buffer_put_lines..cSsg|] }t|qSrirNrririrjr~sz$buffer_put_lines..roN)rrr)r`r?ririrjbuffer_put_liness r@)NN)N) NrZNNrPrTNN)r)rN)rN)rF)rPNN)r)r)rF)rf)rT __future__r contextlibrcsvrrrr functoolsriorr4rshutilr typingr r r r rrrrrrrr unicodedatarnumpyrZpandas._config.configrrZ pandas._libsrZpandas._libs.missingrZpandas._libs.tslibsrrrrrr Zpandas._libs.tslibs.nattyper!Zpandas._typingr"r#r$r%r&r'r(r)r*r+r,Zpandas.core.dtypes.commonr-r.r/r0r1r2r3r4r5r6r7r8Zpandas.core.dtypes.dtypesr9Zpandas.core.dtypes.missingr:r;Zpandas.core.arraysr<r=r>Zpandas.core.arrays.string_r?Zpandas.core.baser@Zpandas.core.commoncorecommonrZpandas.core.constructionrAZpandas.core.indexes.apirBrCrDrEZpandas.core.indexes.datetimesrFZpandas.core.indexes.timedeltasrGZpandas.core.reshape.concatrHZpandas.io.commonrIrJrrKZpandasrLrMrN__annotations__Z_VALID_JUSTIFY_PARAMETERSrXrYrrrrrrrrVrarrvrrrrrrrrrrrrrrr7rrrrrr9r>r@riririrjs      8     48         ; @!'!&E@(&Rp- F  1" $a;+