U
|
¬ý°dx) ã @ sÞ d Z ddlmZ ddlZddlZddlmZmZm Z m
|
Z
|
m Z m Z ddl ZddlmZ ddlmZmZmZmZmZmZ ddlmZ ddlm Z m Z m Z m Z dd l m!Z! dd
|
l"m#Z# dd l$m%Z% erÌdd l&m'Z' G d d dZ(dS )z3
|
Module for formatting output data into CSV files.
|
é )Ú annotationsN)Ú TYPE_CHECKINGÚAnyÚHashableÚIteratorÚSequenceÚcast)Úwriters)ÚCompressionOptionsÚFilePathÚFloatFormatTypeÚ
|
IndexLabelÚStorageOptionsÚ WriteBuffer)Úcache_readonly)ÚABCDatetimeIndexÚABCIndexÚ ABCMultiIndexÚABCPeriodIndex)Únotna)ÚIndex)Ú
|
get_handle)ÚDataFrameFormatterc @ sô e Zd ZU ded< d]d d dddddddddddddddddddZedddd Zed dd d Zeddd d!Zed"dd#d$Z eddd%d&Z
|
dd'd(d)d*Z d+dd,d-Z d+dd.d/Z d+dd0d1Zddd2d3d4Zeddd5d6Zddd7d8d9Zdd:d;d<d=Zed>dd?d@ZedAddBdCZed:ddDdEZedddFdGZedddHdIZed+ddJdKZedLddMdNZdddOdPZdddQdRZ dddSdTZ dUddVdWZ dddXdYZ d:d:ddZd[d\Z dS )^Ú CSVFormatterz
|
np.ndarrayÚcolsÚ ú,NÚwÚstrictÚinferÚ
|
ú"Tr z0FilePath | WriteBuffer[str] | WriteBuffer[bytes]ÚstrzSequence[Hashable] | NonezIndexLabel | Nonez
|
str | Noner
|
z
|
int | NoneÚboolr ÚNone)Ú formatterÚ path_or_bufÚsepr Ú index_labelÚmodeÚencodingÚerrorsÚ compressionÚquotingÚlineterminatorÚ chunksizeÚ quotecharÚ date_formatÚ doublequoteÚ
|
escapecharÚstorage_optionsÚreturnc C s || _ | j j| _|| _|| _| | _|| _|| _|| _| |¡| _
|
|| _ |
|
pNt j | _| | ¡| _|| _|| _| prtj| _|| _| |¡| _| | ¡| _d S ©N)ÚfmtÚframeÚobjÚfilepath_or_bufferr* r, r) r4 r' Ú_initialize_index_labelr( r+ ÚcsvlibÚ QUOTE_MINIMALr- Ú_initialize_quotecharr0 r2 r3 ÚosÚlinesepr. r1 Ú_initialize_columnsr Ú_initialize_chunksizer/ )Úselfr% r& r' r r( r) r* r+ r, r- r. r/ r0 r1 r2 r3 r4 © rD úMd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/io/formats/csvs.pyÚ__init__2 s$
|
zCSVFormatter.__init__)r5 c C s | j jS r6 )r7 Úna_rep©rC rD rD rE rG \ s zCSVFormatter.na_repzFloatFormatType | Nonec C s | j jS r6 )r7 Ú float_formatrH rD rD rE rI ` s zCSVFormatter.float_formatc C s | j jS r6 )r7 ÚdecimalrH rD rD rE rJ d s zCSVFormatter.decimalzbool | Sequence[str]c C s | j jS r6 )r7 ÚheaderrH rD rD rE rK h s zCSVFormatter.headerc C s | j jS r6 )r7 ÚindexrH rD rD rE rL l s zCSVFormatter.indexr )r( r5 c C s6 |dk r2|d kr| ¡ S t|tttjtfs2|gS |S )NF)Ú_get_index_label_from_objÚ
|
isinstanceÚlistÚtupleÚnpÚndarrayr ©rC r( rD rD rE r; p s z$CSVFormatter._initialize_index_labelzSequence[Hashable]c C s" t | jjtr| ¡ S | ¡ S d S r6 )rN r9 rL r Ú_get_index_label_multiindexÚ_get_index_label_flatrH rD rD rE rM y s z&CSVFormatter._get_index_label_from_objc C s dd | j jjD S )Nc S s g | ] }|pd qS ©r rD )Ú.0ÚnamerD rD rE Ú
|
<listcomp> s z<CSVFormatter._get_index_label_multiindex.<locals>.<listcomp>)r9 rL ÚnamesrH rD rD rE rT s z(CSVFormatter._get_index_label_multiindexc C s | j jj}|d krdgS |gS )Nr )r9 rL rX rS rD rD rE rU s
|
z"CSVFormatter._get_index_label_flat)r0 r5 c C s | j tjkr|S d S r6 )r- r< Ú
|
QUOTE_NONE)rC r0 rD rD rE r> s z"CSVFormatter._initialize_quotecharc C s t t| jjtS r6 )r# rN r9 Úcolumnsr rH rD rD rE Úhas_mi_columns s zCSVFormatter.has_mi_columns)r r5 c C sp | j r|d k rd}t||d k rZt|tr<|jf | j}nt|}| jjd d
|f | _| jj }|jf | jS )Nz4cannot specify cols with a MultiIndex on the columns)
|
r] Ú TypeErrorrN r Ú_format_native_typesÚ_number_formatrO r9 Úlocr\ )rC r ÚmsgZnew_colsrD rD rE rA s
|
z CSVFormatter._initialize_columnsÚint)r/ r5 c C s& |d kr dt | jpd p dS t|S )Ni é )Úlenr rc )rC r/ rD rD rE rB £ s z"CSVFormatter._initialize_chunksizezdict[str, Any]c C s | j | j| j| j| jdS )z7Dictionary used for storing number formatting settings.©rG rI r1 r- rJ rf rH rD rD rE r` ¨ s ûzCSVFormatter._number_formatr c sN j j}t|ttfr8 jd k r8t fdd|D }nt|trJ| ¡ }|S )Nc s$ g | ] }t |r | j¡nd qS rV )r Ústrftimer1 )rW ÚxrH rD rE rY » s z+CSVFormatter.data_index.<locals>.<listcomp>) r9 rL rN r r r1 r r Zremove_unused_levels)rC Ú
|
data_indexrD rH rE ri ³ s ÿþÿ
|
zCSVFormatter.data_indexc C s | j rt| jddS dS d S )NÚnlevelsrd r )rL Úgetattrri rH rD rD rE rj Á s zCSVFormatter.nlevelsc C s t | jtttjtfS r6 )rN rK rP rO rQ rR r rH rD rD rE Ú _has_aliasesÈ s zCSVFormatter._has_aliasesc C s t | jp | jS r6 )r# rl rK rH rD rD rE Ú_need_to_save_headerÌ s z!CSVFormatter._need_to_save_headerc C sf | j rRt| jtrtt| jt| jkrLtdt| j dt| j d| jS tt t
|
| jS d S )NzWriting z cols but got z aliases) rl rN rK r# ÚAssertionErrorre r Ú
|
ValueErrorr r r rH rD rD rE Ú
|
write_colsÐ s ÿzCSVFormatter.write_colszlist[Hashable]c C sH g }| j r*| jr*t| jts tt| j}| jr6| jrD|t| j7 }|S r6 ) rL r( rN r rn rO r] rl rp )rC Úencoded_labelsrD rD rE rq Þ s
|
zCSVFormatter.encoded_labelsc
|
C s` t | j| j| j| j| j| jd6}tj|j | j
|
| j | j | j | j| jd| _| ¡ W 5 Q R X dS )z+
|
Create the writer & save.
|
)r* r+ r, r4 )r. Ú delimiterr- r2 r3 r0 N)r r: r) r* r+ r, r4 r< ÚwriterÚhandler. r' r- r2 r3 r0 Ú_save)rC ZhandlesrD rD rE Úsaveë s& úù
|
zCSVFormatter.savec C s | j r| ¡ | ¡ d S r6 )rm Ú _save_headerÚ
|
_save_bodyrH rD rD rE ru s zCSVFormatter._savec C s: | j r | jr | j | j¡ n| ¡ D ]}| j |¡ q$d S r6 )r] rl rs Úwriterowrq Ú _generate_multiindex_header_rows)rC ÚrowrD rD rE rw
|
s zCSVFormatter._save_headerzIterator[list[Hashable]]c c s¬ | j j}t|jD ]h}g }| jrd| |j| ¡ t| jt rdt
|
| jdkrd| dgt
|
| jd ¡ | | |¡¡ |V q| j r¨t| j dhkr¨| j dgt
|
| V d S )Nrd r )r9 r\ Úrangerj rL ÚappendrZ rN r( rO re ÚextendZ_get_level_valuesrq Úset)rC r\ ÚiZcol_linerD rD rE rz s z-CSVFormatter._generate_multiindex_header_rowsc C s\ t | j}|| j d }t|D ]6}|| j }t|| j |}||krJ qX| ||¡ q d S )Nrd )re ri r/ r| ÚminÚ _save_chunk)rC ZnrowsÚchunksr Ústart_iÚend_irD rD rE rx & s
|
|
zCSVFormatter._save_body)r r
r5 c sr t ||}| jj| }|jjf | j fddtt jD }| j | j
|
f | j}t ||| j | j| j¡ d S )Nc s g | ]} |¡qS rD )Z iget_values)rW r ©ÚresrD rE rY 6 s z,CSVFormatter._save_chunk.<locals>.<listcomp>)Úslicer9 ZilocZ_mgrZto_native_typesr` r| re Úitemsri r_ Ú
|
libwritersZwrite_csv_rowsrj r rs )rC r r
ZslicerZdfÚdataÚixrD r rE r 0 s
|
ûzCSVFormatter._save_chunk)r r NNr Nr r Nr Nr! NTNN)!Ú__name__Ú
|
__module__Ú __qualname__Ú__annotations__rF ÚpropertyrG rI rJ rK rL r; rM rT rU r> r] rA rB r` r ri rj rl rm rp rq rv ru rw rz rx r rD rD rD rE r / sr
|
î2*
|
|
r ))Ú__doc__Ú
|
__future__r Úcsvr< r? Útypingr r r r r r ÚnumpyrQ Z pandas._libsr r Zpandas._typingr
|
r r r r r Zpandas.util._decoratorsr Zpandas.core.dtypes.genericr r r r Zpandas.core.dtypes.missingr Zpandas.core.indexes.apir Zpandas.io.commonr Zpandas.io.formats.formatr r rD rD rD rE Ú<module> s
|
