a  d)@sUdZddlmZddlmZmZmZmZmZddl m Z ddl m Z m Z ddlmZddlmZddlmZdd lZdd lZdd lmZdd lmZmZmZmZmZmZdd lm Z dd l!Z"dd l#m$Z$m%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0ddl1m2Z2ddl3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9ddl:m;Z;mm?Z?m@Z@ddlAmBZBmCZCmDZDddlEmFZFddlGmHZHdd lImJmKZLddlMmNZNddlOmPZPmQZQmRZRmSZSddlTmUZUddlVmWZWddlXmYZYddlZm[Z[m\Z\ddl]m^Z^erBddl_m`Z`maZambZbmcZcmdZdmeZemfZfmgZgmhZhmiZimjZjdd lkmlZlmmZmd!Znd"eod#<d$Zpd%Zqd"eod&<Gd'd(d(ZrGd)d*d*ZsGd+d,d,ZtGd-d.d.etZud,d/d0d1Zvd2d/d3d4Zwd2d/d5d6ZxGd7d8d8ZyGd9d:d:Zzdd;dd?d@Z{e ddz2CategoricalFormatter.to_string..rf[]rh)rXrir\rlrrjoinappendrV)rarX fmt_valuesvaluesresultr\rbrbrc to_strings   zCategoricalFormatter.to_string)NTrST)__name__ __module__ __qualname__rdrlrrrrbrbrbrcrRs rRc@sxeZdZddddddd dd dd d d d ddZd dddZd dddZddddZddddZd dddZdS)SeriesFormatterNTrSFrHrT bool | strrUrV str | None int | NonerW) seriesrYrZheaderindexr[namerndtypemax_rowsmin_rowsr]c Cst||_|dur|nt|_||_||_||_||_||_| |_| |_ |durTt d}||_ | |_ t |_|dS)Ndisplay.float_format)rr rYrr[rrZrrrrrnrget_adjustmentadj _chk_truncate) rarrYrZrrr[rrnrrrrbrbrcrd szSeriesFormatter.__init__recCs||j}|j}|o t|j|k}|j}|rtt|}|rDt||}|dkr`|}|jd|}n*|d}t|jd||j| df}||_ nd|_ ||_ ||_ dS)N) rrrirrintminilocr8 tr_row_num tr_seriesis_truncated_vertically)rarrrrrow_numrbrbrcr)s$  "zSeriesFormatter._chk_truncatecCsH|jj}d}t|jjdddurLt|jjtttfs8J|d|jjj7}|jdur|dur|rj|d7}t j |dd}|d|7}|j d us|j d kr|j r|r|d7}|d t |j7}|jdur|jdurt|jjd d}|r|r|d7}|d t |7}t|jjtr@|jj}|r8|d7}||7}t|S)Nr_freqzFreq: Frf  rh) escape_charszName: Ttruncatergrzdtype: rh)rrgetattrr isinstancer6r4r7Zfreqstrr; pprint_thingrZrrirrr'_valuesrjrV)rarr\Z series_nameZ dtype_namerkrbrbrcrlDs@  zSeriesFormatter._get_footerztuple[list[str], bool]cCsR|jj}t|tr4tdd|jD}|jdd}n|jdu}|jdd}||fS)Ncss|] }|VqdSNrbrurrbrbrc qrxz7SeriesFormatter._get_formatted_index..T)names)r)rrrr3anyrformatr)rar have_header fmt_indexrbrbrc_get_formatted_indexms   z$SeriesFormatter._get_formatted_indexrmcCst|jjd|j|j|jdS)N)rnr[ leading_space)rorrrnr[rrqrbrbrcrrxsz%SeriesFormatter._get_formatted_valuesc Cs2|j}|}t|dkr2t|jjd|dS|\}}|}|jrd}|j }t t |}|j ||d}|dkrd} nd} |j j | g|dd d} |||| ||dd |jr|j jdg|dd|gR} n|j d|} |jr|r|dd | } |r$| d |7} td | S) Nrz([], )r...z..rLmoder_rh)rrlrityperrrrrrrrrrrMinsertradjoinrrVr{) rarr\rrr}Z n_header_rowsrwidthZdot_strrrbrbrcrs2   " zSeriesFormatter.to_string) NTTTrSFNTNN) rrrrdrrlrrrrrbrbrbrcrs & )  rc@sPeZdZddddZddddd Zdd ddd d ddZdddddZdS)TextAdjustmentrWrecCstd|_dS)Nzdisplay.encoding)rencodingrqrbrbrcrdszTextAdjustment.__init__rVrtextr]cCst|SrrirarrbrbrcriszTextAdjustment.lenrKrrmtextsmax_lenrr]cCstj|||dS)Nr)r;rMrarrrrbrbrcrMszTextAdjustment.justify)spacer]cOs"tj|g|R|j|jd|S)N)ZstrlenZjustfunc)r;rrirM)rarlistskwargsrbrbrcrszTextAdjustment.adjoinN)rK)rrrrdrirMrrbrbrbrcrsrcsHeZdZddfdd Zddddd Zdd ddd d ddZZS)EastAsianTextAdjustmentrWrecs6ttdrd|_nd|_dddddd|_dS)Nz!display.unicode.ambiguous_as_widerr)ZNaNWFH)superrdrambiguous_width_EAW_MAPrq __class__rbrcrds  z EastAsianTextAdjustment.__init__rVrrcs(t|tst|Stfdd|DS)zN Calculate display width considering unicode East Asian Width c3s"|]}jt|jVqdSr)rgetrr)rucrqrbrcrsz.EastAsianTextAdjustment.len..)rrVrisumrrbrqrcris   zEastAsianTextAdjustment.lenrKz Iterable[str]rmrcsXfdd|dkr(fdd|DS|dkrBfdd|DSfdd|DSdS) Ncs|t|Srr)t)rrarbrc_get_padsz1EastAsianTextAdjustment.justify.._get_padrJcsg|]}||qSrb)ljustruxrrbrcrwrxz3EastAsianTextAdjustment.justify..rLcsg|]}||qSrb)rLrrrbrcrwrxcsg|]}||qSrb)rjustrrrbrcrwrxrbrrb)rrrarcrMs zEastAsianTextAdjustment.justify)rK)rrrrdrirM __classcell__rbrbrrcrs  rrecCstd}|rtStSdS)Nz display.unicode.east_asian_width)rrr)Zuse_east_asian_widthrbrbrcrsrzdict[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_rbrbrcget_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)rrrrrZ)rr)rheightrrrbrbrcget_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.r_NTrSF.rGz Axes | NonezColspaceArgType | Nonezbool | list[str]rUrVzFormattersType | NonerFloatFormatType | None bool | NonerrrW)framecolumns col_spacerrr[ formattersrMrnsparsify index_namesrrrrdecimal bold_rowsescaper]cCs||_|||_|||_||_||_||_|||_ | ||_ | |_ | | |_| |_||_||_||_| |_| |_||_||_||_||_|j|_|t|_dSr)r_initialize_columnsr_initialize_colspacerrrr[_initialize_formattersr_initialize_justifyrMrn_initialize_sparsifyrshow_index_namesrrrrrrr_calc_max_cols_fittedmax_cols_fitted_calc_max_rows_fittedmax_rows_fittedtr_framerrr)rarrrrrr[rrMrnrrrrrrrrrrbrbrcrd4s.       zDataFrameFormatter.__init__zlist[list[str]]recCs*|}|jr&||j}|d||S)zP Render a DataFrame to a list of columns (as lists of strings). r)_get_strcols_without_indexrrrr)rastrcolsZ str_indexrbrbrc get_strcolscs   zDataFrameFormatter.get_strcolscCs|jdup|jdko|jS)NTr)r is_truncatedrqrbrbrcshould_show_dimensionsos z)DataFrameFormatter.should_show_dimensionscCst|jp |jSr)rUis_truncated_horizontallyrrqrbrbrcruszDataFrameFormatter.is_truncatedcCst|jot|j|jkSr)rUrrirrqrbrbrcrysz,DataFrameFormatter.is_truncated_horizontallycCst|jot|j|jkSr)rUrrirrqrbrbrcr}sz*DataFrameFormatter.is_truncated_verticallycCs dt|jdt|jjdS)Nz [z rows x z columns])rirrrqrbrbrcdimensions_infosz"DataFrameFormatter.dimensions_infocCs t|jjSr) _has_namesrrrqrbrbrchas_index_namessz"DataFrameFormatter.has_index_namescCs t|jjSr)rrrrqrbrbrchas_column_namessz#DataFrameFormatter.has_column_namescCst|j|j|jfSr)allrrrrqrbrbrcshow_row_idx_namessz%DataFrameFormatter.show_row_idx_namescCst|j|j|jfSr)r r rrrqrbrbrcshow_col_idx_namessz%DataFrameFormatter.show_col_idx_namesrcCst|jpt|jt|jSr)rrrirrqrbrbrcmax_rows_displayedsz%DataFrameFormatter.max_rows_displayed)rr]cCs|durtdS|S)Nzdisplay.multi_sparser)rarrbrbrcrsz'DataFrameFormatter._initialize_sparsifyrC)rr]cCsT|dur iSt|jjt|ks*t|tr.|Stdt|dt|jjddS)NzFormatters length(+) should match DataFrame number of columns(r)rirrrdict ValueError)rarrbrbrcrs  z)DataFrameFormatter._initialize_formatters)rMr]cCs|durtdS|SdS)Nzdisplay.colheader_justifyr)rarMrbrbrcrsz&DataFrameFormatter._initialize_justifyr2)rr]cCs,|dur t|}|j||_|S|jjSdSr)r5rr)rarcolsrbrbrcrs  z&DataFrameFormatter._initialize_columnsr?)rr]csduri}ntttfrBdi}|fdd|jjDnttrD]&}||jjvrT|dkrTtd|qT}nHt |jjt krtdt dt |jjdt t |jj}|S)Nr_csi|] }|qSrbrb)rucolumnrrbrc rxz;DataFrameFormatter._initialize_colspace..z,Col_space is defined for an unknown column: zCol_space length(rr) rrrVupdaterrrkeysrrirzip)rarrrrbrrcrs(    z'DataFrameFormatter._initialize_colspacecCs0|s|jSt\}}||r&|S|jSdS)z%Number of columns fitting the screen.N)_is_in_terminalrr_is_screen_narrow)rarrrbrbrcrs   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)rrr_get_number_of_auxiliary_rows_is_screen_short_adjust_max_rows)rarrrrbrbrcrs    z(DataFrameFormatter._calc_max_rows_fitted)rr]cCs(|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 )rirrr)rarrbrbrcrs z#DataFrameFormatter._adjust_max_rowscCst|jdkp|jdkS)z/Check if the output is to be shown in terminal.r)rUrrrqrbrbrcrsz"DataFrameFormatter._is_in_terminalcCst|jdkot|jj|kSNr)rUrrirr)ra max_widthrbrbrcrsz$DataFrameFormatter._is_screen_narrowcCst|jdkot|j|kSr)rUrrir)raZ max_heightrbrbrcrsz#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)rrir splitlinesr)raZdot_rowZ prompt_rowZnum_rowsrbrbrcrsz0DataFrameFormatter._get_number_of_auxiliary_rowscCs |jr||jr|dS)zY Check whether the frame should be truncated. If so, slice the frame up. N)r_truncate_horizontallyr_truncate_verticallyrqrbrbrcrszDataFrameFormatter.truncatecCs|jdusJ|jd}|dkr|jjddd|f}|jjdd| df}t||fdd|_t|jttfrg|jd||j| d|_n&tt |j }|jjddd|f|_||_ dS)zRemove columns, which are not to be displayed and adjust formatters. Attributes affected: - tr_frame - formatters - tr_col_num NrrZaxis) rrrr8rrlisttuplerrrZ tr_col_num)raZcol_numrJrKrbrbrcr!s   z)DataFrameFormatter._truncate_horizontallycCs|jdusJ|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)rrrr8rrrr)rarheadtailrbrbrcr"9s  z'DataFrameFormatter._truncate_verticallyc sg}tjsbjsbtjD]>\}}|}t|jtj |dj d}| |q|Stjrt t tj_tjtjkrtdtjdtjdddjD}n j}jr|D]}| dqtjD]\}}||}ttj |dgfd d |DR}|}t|j|j d }tgfd d |D|R} j j|| jd }| ||q|S)Nr)stringsrMminimumrzWriting z cols but got z aliasescSsg|] }|gqSrbrb)rulabelrbrbrcrwbrxzADataFrameFormatter._get_strcols_without_index..r_c3s|]}j|VqdSrrrirrqrbrcrmrxz@DataFrameFormatter._get_strcols_without_index..)r)rc3s|]}j|VqdSrr+rrqrbrcrtrxr)r$r enumerater format_col_make_fixed_widthrMrrrrr|rr$rVrirr_get_formatted_column_labelsr max) rarrvrr} str_columnsrZcheaderZheader_colwidthrrbrqrcrKsP        z-DataFrameFormatter._get_strcols_without_indexrm)rvr]c CsL|j}||}t|jdd|fj||j|j|j|j ||j |j dS)N)rnr[rrr) r_get_formatterrorrrnr[rrrrr)rarvr formatterrbrbrcr-zs zDataFrameFormatter.format_col str | intCallable | NonecCs`t|jttfr2t|r,tt|}|j|SdSn*t|rN||jvrN|j|}|j|dSdSr) rrr$r%r#rrrr)rarvrbrbrcr2s   z!DataFrameFormatter._get_formatter)rr]csddlm}|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)rulevelrbrbrcrrxzBDataFrameFormatter._get_formatted_column_labels..cs"|jvr|rsd|S|S)N )r)ry) need_leadsprestrict_formattingrarbrc space_formatszEDataFrameFormatter._get_formatted_column_labels..space_formatc3s"|]fddDVqdS)csg|]}|qSrbrb)rur9)r<rrbrcrwrxzMDataFrameFormatter._get_formatted_column_labels...Nrb)ru)r<rrcrrxcSsg|] }t|qSrb)r$rrbrbrcrwrxzCDataFrameFormatter._get_formatted_column_labels..cs0g|](\}}|s&|r&d|n|gqS)r8)r2)rurvr)r:rarbrcrws)Zpandas.core.indexes.multir6rrr3rr$rrdtypesrrlevelsrmapr%rrir,)rarr6rZ fmt_columnsr>r1rb)r:r;rar<rcr/s,      z/DataFrameFormatter._get_formatted_column_labelscsddjD|j}|j}d}t|trL|jjdj |d}n|jj |dg}fdd|D}j j d g|R d }j rd dD}n d g|j}jr||S|SdS) NcSsi|]\}}|tt|qSrb)rr)rukvrbrbrcrrxz;DataFrameFormatter._get_formatted_index.. __index__F)rrrr3)rr3c s.g|]&}ttt|dddjdqS)rJr_r)rMr)r)r%r.r$rrrrrarbrcrws z;DataFrameFormatter._get_formatted_index..rrhcSsg|] }t|qSrbrVrrbrbrcrwrxr_)ritemsrrr2rr3rrr rrsplitr _get_column_name_listZnlevelsr)rarrrfmtrZadjoinedZ col_headerrbrDrcrs,     z'DataFrameFormatter._get_formatted_indexzlist[Hashable]cCsLg}|jj}t|tr.|dd|jDn||jdur@dn|j|S)Ncss|]}|durdn|VqdSr^rbrrbrbrcrrxz;DataFrameFormatter._get_column_name_list..r_)rrrr3extendrr|r)rarrrbrbrcrHs  z(DataFrameFormatter._get_column_name_list)NNTTrSNNNNTNNNFrFT)(rrr__doc__rIrQrdrpropertyrrrrrrr r r r rrrrrrrrrrrrrr!r"rr-r2r/rrHrbrbrbrcr.s~  4/    /  )(rc@seZdZdZdddddZd$d d d d d d d d dddZd%d d dd dddZd&dd dddddddd dd d d dd 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. rrW)rIr]cCs ||_dSr)rI)rarIrbrbrcrdszDataFrameRenderer.__init__NF"FilePath | WriteBuffer[str] | Nonerzstr | list | tuple | NonerUzint | bool | None)rYrclassesnotebookbordertable_id render_linksr]c 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)rOrQrRrSrYr)Zpandas.io.formats.htmlrTrUrIrsave_to_buffer) rarYrrOrPrQrRrSrTrUZKlassZhtml_formatterstringrbrbrcto_htmls" zDataFrameRenderer.to_htmlr)rYrrr]cCs0ddlm}||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)rrV)Zpandas.io.formats.stringrZrIrrW)rarYrrrZZstring_formatterrXrbrbrcr5s zDataFrameRenderer.to_string,winfer"Tstrictz7FilePath | WriteBuffer[bytes] | WriteBuffer[str] | NonerVzSequence[Hashable] | NonezIndexLabel | Noner@zStorageOptions | None) path_or_bufrsepr index_labelr compressionr` quotecharlineterminator chunksize date_format doublequote escapecharerrorsstorage_optionsr]cCsddlm}|dur d}t}nd}||| ||||||||| | | | |||jd}||r~t|tsjJ|}||SdS)z; Render dataframe as comma-separated file. r) CSVFormatterNTF)r`rerarrjrcr`rrbrrfrdrgrhrirkr3)Zpandas.io.formats.csvsrlr rIsavergetvalueclose)rar`rrarrbrrcr`rdrerfrgrhrirjrkrlZcreated_bufferZ csv_formattercontentrbrbrcto_csvOs> zDataFrameRenderer.to_csv)NNNFNNF)NNN)NNr[NNr\r]Nr^NNNTNr_N)rrrrKrdrYrrqrbrbrbrcrMs>5rMrVrNr)rXrYrr]cCsZt||d:}|||dur6|WdSWddS1sL0YdS)zQ Perform serialization. Write to buf or return as string if buf is None. )rN) get_bufferwritern)rXrYrfrbrbrcrWs  rWzIGenerator[WriteBuffer[str], None, None] | Generator[StringIO, None, None])rYrr]ccs|durt|}nt}|dur&d}nt|ts8tdt|drJ|VnVt|trtt|t|d|dd}|VWdq1s0YntddS) 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.rsr\r_)rnewlinez1buf is not a file name and it has no write method) r:r rrVrhasattrr9open TypeError)rYrrtrbrbrcrrs     &rrrSrKrTrr5rrzstr | int | Nonerrm) r~r3rnr[digitsrrMrrr`fallback_formatterr]c Cst|jdrt} ndt|jtr&t} nRt|jdr:t} n>t|jtrLt } n,t|jdr`t } nt|jdrtt } nt } |durd}|durt d}|durt d}| |||||||||| | 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] MmZfciuN rdisplay.precision) ryr[rnr3rrMrrr`rz)rZ is_np_dtyperDatetime64Formatterrr(Datetime64TZFormatterTimedelta64Formatterr)ExtensionArrayFormatterFloatArrayFormatterIntArrayFormatterGenericArrayFormatterr get_result) r~r3rnr[ryrrMrrr`rzZ fmt_klassZfmt_objrbrbrcrosB)  roc@sPeZdZddd d d d d d d dddd dd ddZddddZddddZdS)rNrSr~rKrTrrr5rVr4rrrUrrW) r~ryr3r[rrnrMrr` fixed_widthrrzr]c CsL||_||_||_||_||_||_||_||_| |_| |_ | |_ | |_ dSr) r~ryr[rr3rnrMrr`rrrz) rar~ryr3r[rrnrMrr`rrrzrbrbrcrdszGenericArrayFormatter.__init__rmrecCs|}t||jSr)_format_stringsr.rM)rar}rbrbrcr0sz GenericArrayFormatter.get_resultc sjdur0td}|dur6tdfdd}nj}jdurHjn6jdurZjn$jduoljtk}ttjd|dfdd}t j d d }t |t j std t|t}|t jt|ttd t|jd @}j}|dur|}g}t|D]\} } || r jdur<|r<|d|| nD|| rV||| n*|durfd} nd} || j|| dq|S)Nrrcst|dddS)Nz .drt)_trim_zeros_single_floatr=) precisionrbrc9sz7GenericArrayFormatter._format_strings..r)r quote_stringsc sjdurtt|rtt|rtz<|dur*WdS|tur._formatTZ extract_numpyz.base_formattercst|rt|SjSr)r+rVr[rrqrbrcrsrcs|djdS)Nrr)replacerr)rrarbrcdecimal_formattersz?FloatArrayFormatter._value_formatter..decimal_formattercs0t|r&t|kr|SdSnjSdS)Ng)r+absr[r)rrarrbrcr3s   z7FloatArrayFormatter._value_formatter..formatter)rnr)rarnrr3rb)rrrnrarrc_value_formatters   z$FloatArrayFormatter._value_formatter np.ndarrayc slddddddddddddjd ur@jjjSjrPtd nd fd d }jd urjrjd urd}nd}t|jj d}qj}n fdd}||}js|St |dkrt dd|D}|j dk}nd}t j}|dk}|dj k|dk@} | s:|rh|rhjd urLd}nd}t|jj d}||}|S)z Returns the float values converted into strings using the parameters given at initialisation, as a numpy array r<rrV)r~r3r[cs<t|}tfddt||D|j}|S)Ncs g|]\}}|s|nqSrbrb)ruvalr|r3r[rbrcrwszWFloatArrayFormatter.get_result_as_array..format_with_na_rep..)r*rarrayrravelreshaper)r~r3r[mask formattedrbrrcformat_with_na_reps zCFloatArrayFormatter.get_result_as_array..format_with_na_repcSst|}t|}t|t|}}g}t|||||D]\}} } } } | sl| sl|||qF| s||| d|dqF| s|| } | dr||| dq||d| dqF||d|dqFt | |j S)N+j-) rrealrimagr*rr|rs startswithrrr)r~r3r[Z real_valuesZ imag_valuesZ real_maskZ imag_maskZ formatted_lstrZreal_valZimag_valZre_isnaZim_isnaZimag_formattedrbrbrcformat_complex_with_na_reps,  zKFloatArrayFormatter.get_result_as_array..format_complex_with_na_repNzdisplay.chop_thresholdcs|}jdkr djnj}j}t|}|rF|||}n |||}jr|rjt|j}n t|j}t j |ddS|S)NrJr8objectr) rrMr[r~r!r_trim_zeros_complexr_trim_zeros_floatrasarray)rnr3r[r~Z is_complexrrrrarrbrcformat_values_withs   zCFloatArrayFormatter.get_result_as_array..format_values_withTz{value: .{digits:d}f}z{value:.{digits:d}f})rycs j|Sr)rnrrqrbrcr&rxz9FloatArrayFormatter.get_result_as_array..rcss|]}t|VqdSrrrrbrbrcr2rxz:FloatArrayFormatter.get_result_as_array..Fg.A z{value: .{digits:d}e}z{value:.{digits:d}e})r3r~r[rrrnrr rryrir0rrr) rarZfmt_strrnZformatted_valuesmaxlenZtoo_longZabs_valsZhas_large_valuesZhas_small_valuesrbrrcget_result_as_arraysB          z'FloatArrayFormatter.get_result_as_arrayrmcCs t|Sr)r$rrqrbrbrcrHsz#FloatArrayFormatter._format_strings)NN)rrrrdrrrrrbrbrrcrs8rc@seZdZddddZdS)rrmrecs>|jdurdd}ndd}|jp$|fdd|jD}|S)NFcSs|dj|dS)Nrr=rr=rbrbrcrOrxz3IntArrayFormatter._format_strings..cSs|dj|dS)Nz dr=rr=rbrbrcrQrxcsg|] }|qSrbrbrr3rbrcrwSrxz5IntArrayFormatter._format_strings..)rr3r~)raZ formatter_strr}rbrrcrMs    z!IntArrayFormatter._format_stringsNrrrrrbrbrbrcrLsrcs8eZdZd dddddfdd Zd d d d ZZS)rrNz3np.ndarray | Series | DatetimeIndex | DatetimeArrayrVrW)r~nat_reprgr]c s$tj|fi|||_||_dSr)rrdrrg)rar~rrgrrrbrcrdXszDatetime64Formatter.__init__rmrecsZj}t|tst|}jdur>tjr>fdd|DS|jjjjd}| S)z&we by definition have DO NOT have a TZNcsg|]}|qSrbrrrqrbrcrwkrxz7Datetime64Formatter._format_strings..)r[rg) r~rr6r3r_dataZ_format_native_typesrrgtolist)rar~r}rbrqrcrcs z#Datetime64Formatter._format_strings)rNrrrrdrrrbrbrrcrWs rc@seZdZddddZdS)rrmrec Cs|t|jdd}|j}d}|dur,|jdd}t|tr@|}n t|}t |||j |j |j |j |j|j|j|j|d }|S)NTr)Zboxed) rnr[ryrrMrrr`rz)r1r~r3 _formatterrr,rprrrornr[ryrrMrrr`)rar~r3rzrr}rbrbrcrts,     z'ExtensionArrayFormatter._format_stringsNrrbrbrbrcrssrznp.ndarray | Sequence[float]) percentilesr]c Cs@t|}t|r.t|dkr.t|dks6tdd|}|t}t||}t|rz|t }dd|DSt |}|ddkr|dnd}|ddkrd|dnd}t t t tj|||d t }td|}tj|td }||tt ||<|||t ||<d d|DS) a Outputs rounded and formatted percentiles. Parameters ---------- percentiles : list-like, containing floats from interval [0,1] Returns ------- formatted : list of strings Notes ----- Rounding precision is chosen so that: (1) if any two elements of ``percentiles`` differ, they remain different after rounding (2) no entry is *rounded* to 0% or 100%. Any non-integer is always rounded to at least 1 decimal place. Examples -------- Keeps all entries different after rounding: >>> 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%'] rrz/percentiles should all be in the interval [0,1]dcSsg|] }|dqS%rbrtrbrbrcrwrxz&format_percentiles..N)to_beginto_endrcSsg|] }|dqSrrbrtrbrbrcrwrx)rrr%r rroundastyperiscloserVuniquefloorlog10rZediff1dr0Z empty_liker)rZpercentiles_round_typeZint_idxoutZ unique_pctsrrprecrbrbrcformat_percentiless6"        rz2np.ndarray | DatetimeArray | Index | DatetimeIndexrUr~r]cCs~t|ts|}t|ttfs(t|}|jdur6dS|j}|tk}t|j }t |}t |||dk dk}|rzdSdS)NFrT)rr2rr-r6tzZasi8rrrrr logical_andr)r~ values_intconsider_valuesZresoZppd even_daysrbrbrc is_dates_onlys   rrzNaTType | Timestamp)rrr]cCs|tur |St|Sr)rrV)rrrbrbrc_format_datetime64sr)rrrgr]cCs&t|tr|S|r||S|jSdSr)rr strftimeZ _date_repr)rrrgrbrbrc_format_datetime64_dateonlys   rr)is_dates_only_rrgr]cs"|rfddSfddSdS)z]Return a formatter callable taking a datetime64 as input and providing a string as outputcst|dS)N)rrg)rr=rgrrbrcrsz'get_format_datetime64..cs t|dS)Nr)rr=rrbrcrrxNrb)rrrgrbrrcget_format_datetime64srz*np.ndarray | DatetimeArray | DatetimeIndex)r~rgr]cCs6t|tjr|jdkr|}t|}|r2|p0dS|S)z6given values and a date_format, return a string formatrz%Y-%m-%d)rrrndimrr)r~rgidorbrbrc!get_format_datetime64_from_valuess rc@seZdZddddZdS)rrmrecs@t|j}|jt}|jp(t||jdfdd|D}|S)zwe by definition have a TZ)rgcsg|] }|qSrbrbrrrbrcrw.rxz9Datetime64TZFormatter._format_strings..)rr~rrr3rrg)rarr~r}rbrrcr's  z%Datetime64TZFormatter._format_stringsNrrbrbrbrcr&srcs8eZdZddddddfdd Zd d d d ZZS)rrFznp.ndarray | TimedeltaIndexrVrUrWr~rboxr]c s$tj|fi|||_||_dSr)rrdrr)rar~rrrrrbrcrd4szTimedelta64Formatter.__init__rmrecs.|jpt|j|j|jdfdd|jDS)N)rrcsg|] }|qSrbrbrrrbrcrwCrxz8Timedelta64Formatter._format_strings..)r3get_format_timedelta64r~rrrqrbrrcr?s z$Timedelta64Formatter._format_strings)rFrrbrbrrcr3s 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|dust|rt|rSt|ts.t|}|jd}rJd|d}|S)Nr')r&r*rrZ _repr_base)rrrrrrbrcres   z*get_format_timedelta64.._formatter)viewrint64rrr) r~rrrrZ one_day_nanosZ not_midnightZbothrrrbrrcrFs     rzTextAdjustment | None)r(rMr)rr]cst|dks|dkr|S|dur(tn|tfdd|D|durTt|tddurpkrpdddfdd fd d |D}j||d }|S) Nrr c3s|]}|VqdSrrr) adjustmentrbrcrrxz$_make_fixed_width..rrV)rr]cs6dur2dk|k@r2|ddd}|S)Nrrrr=)rconf_maxrrbrcjustsz_make_fixed_width..justcsg|] }|qSrbrbr)rrbrcrwrxz%_make_fixed_width..r)rirr0rrM)r(rMr)rrrb)rrrrrcr.us r.r) str_complexesrr]c sgg}}|D]@}td|}|d|dd|d|ddqt|}t|||}t|dkrvgStdd|Dd fd d t|d|||dD}|S) z Separates the real and imaginary parts from the complex number, and executes the _trim_zeros_float method on each of those. z([j+-])r_Nrcss|]}t|VqdSrr)rupartrbrbrcrrxz&_trim_zeros_complex..rcs6g|].\}}||d|ddddqS)rrN>rrb)ruZreal_ptZimag_ptZ padded_lengthrbrcrwsz'_trim_zeros_complex..)rerGr|r{rirr0r) rrZ real_partZ imag_partrtrimmednZ padded_partspaddedrbrrcrs    r) str_floatr]cCs |d}|dr|d7}|S)zX Trims trailing zeros after a decimal point, leaving just one if necessary. 0r)rstripendswith)rrbrbrcrs  rnp.ndarray | list[str]) str_floatsrr]csp|}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]*$rUrecst|duSr)rmatchr=) number_regexrbrcis_number_with_decimalsz1_trim_zeros_float..is_number_with_decimalrrcs0fdd|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|qSrbrbrrrbrcrwrxz:_trim_zeros_float..should_trim..rcss|]}|dVqdS)rNrrrbrbrcrrxz9_trim_zeros_float..should_trim..)rir )r~Znumbersrrbrc should_trimsz&_trim_zeros_float..should_trimcs$g|]}|r|ddn|qS)Nrrbrrrbrcrwrxz%_trim_zeros_float..cs*g|]"}|r"|r"|dn|qS)rrr)rrrbrcrws)rcompile)rrrrrrb)rrrrcrs  rr2)rr]cCs$t|trtj|jS|jduSdSr)rr3comZ any_not_nonerr)rrbrbrcrs  rc@s\eZdZdZdddddddd d d d d ddddddZd!ddddddZddddd ZdS)" EngFormatterzl Formats float values according to engineering format. Based on matplotlib.ticker.EngFormatter r9zartprur|r_rAr{GTPEZY)iiiiiiirrr r~NFrrUrWaccuracyuse_eng_prefixr]cCs||_||_dSrrr)rarrrbrbrcrd szEngFormatter.__init__floatrV)numr]c 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 urd }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 rSinfrrrrzE-Z02dzE+rNz{mant: g}{prefix}z{mant: .rz f}{prefix})mantprefix)r rVis_nan is_infinitermathrrrr0 ENG_PREFIXESrrrr) rar!ZdnumsignZpow10Z int_pow10r$r#Z format_strrrbrbrc__call__s4       zEngFormatter.__call__)NF)rrrrKr(rdr*rbrbrbrcr s,r rrrWrcCstdt||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)rr rrbrbrcset_eng_float_formatOs7r+r_zbool | object | strzlist[dict[int, int]])r?sentinelr]c 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)rir,r|) r?r,controlrr7 last_indexlengthsrvkeyrbrbrcget_level_lengthss    r1zWriteBuffer[str])rYlinesr]cCs4tdd|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)rrVrrbrbrcrrxz#buffer_put_lines..cSsg|] }t|qSrbrErrbrbrcrwrxz$buffer_put_lines..rhN)rrsr{)rYr2rbrbrcbuffer_put_liness r3)NN)N) NrSNNrKrTNN)r)rN)rN)rF)rKNN)r)r)rF)r_)rK __future__rcollections.abcrrrrr contextlibrcsvr r rr functoolsr ior r'rshutilrtypingrrrrrr unicodedatarnumpyrZpandas._config.configrrZ pandas._libsrZpandas._libs.missingrZpandas._libs.tslibsrrrrrrZpandas._libs.tslibs.nattyper Zpandas.core.dtypes.commonr!r"r#r$r%r&Zpandas.core.dtypes.dtypesr'r(r)Zpandas.core.dtypes.missingr*r+Zpandas.core.arraysr,r-r.Zpandas.core.arrays.string_r/Zpandas.core.baser0Zpandas.core.commoncorecommonr Zpandas.core.constructionr1Zpandas.core.indexes.apir2r3r4r5Zpandas.core.indexes.datetimesr6Zpandas.core.indexes.timedeltasr7Zpandas.core.reshape.concatr8Zpandas.io.commonr9r:rr;Zpandas._typingr<r=r>r?r@rArBrCrDrErFZpandasrGrHrI__annotations__Z_VALID_JUSTIFY_PARAMETERSrQrRrrrrrrrrMrWrrrorrrrrrrrrrrrrrr.rrrrr r+r1r3rbrbrbrcs                    4 ; @!'!&C (&RpJ E  1"! $`;+