§
    {Øj?  ã            
      óf  — d dl mZ d dlZd dlmZ d dlmZmZ d dlm	Z	 d dl
mZmZmZmZ d dlZd dlmZ d dlmZ d dlmZ d d	lmZ d d
lmZ d dlmZmZmZ d dl m!Z! d dl"m#Z#m$Z$ d dl%m&Z&m'Z' er$d dl(m)Z)m*Z* d dl+m,Z,m-Z-m.Z. d dl%m/Z/ d dl0m1Z1m2Z2m3Z3m4Z4  G d„ de!eeeej5        df         ¦  «        Z6dS )é    )ÚannotationsN)Úreduce)ÚBytesIOÚ
TextIOBase)Úchain)ÚTYPE_CHECKINGÚAnyÚLiteralÚcast©ÚArrowDataFrame©Ú	ArrowExpr)ÚArrowSelectorNamespace©ÚArrowSeries)Úbuild_list_arrayÚcast_to_comparable_string_typesÚchunked_array)ÚEagerNamespace)Úcombine_alias_output_namesÚcombine_evaluate_output_names)ÚImplementationÚcheck_column_names_are_unique)ÚIteratorÚSequence)ÚChunkedArrayAnyÚ
IncompleteÚ	ScalarAny)ÚVersion)ÚCorrelationMethodÚ	IntoDTypeÚNormalizedSourceÚPythonLiteralc                  ó6  — e Zd Zej        ZedFd„¦   «         ZedGd„¦   «         ZedHd„¦   «         Z	dId„Z
ddœdJd„ZdKd„ZdLd„ZdMd„ZdNd!„ZdOd%„ZdOd&„ZdPd'„ZdPd(„ZdPd)„ZdPd*„ZdQd.„ZdQd/„ZdQd0„ZedRd2„¦   «         ZdSd3„ZdPd4„ZdPd5„ZdPd6„Z	 dTdUd=„ZdVdB„ZdWdE„Z d7S )XÚArrowNamespaceÚreturnútype[ArrowDataFrame]c                ó   — t           S ©Nr   ©Úselfs    úŽ/Users/tedpeters/Documents/Projects/Claude Projects/Misc Tools/notename-sheets/.venv/lib/python3.11/site-packages/narwhals/_arrow/namespace.pyÚ
_dataframezArrowNamespace._dataframe.   s   € åÐó    útype[ArrowExpr]c                ó   — t           S r*   r   r+   s    r-   Ú_exprzArrowNamespace._expr2   s   € åÐr/   útype[ArrowSeries]c                ó   — t           S r*   r   r+   s    r-   Ú_serieszArrowNamespace._series6   s   € åÐr/   Úversionr    ÚNonec               ó   — || _         d S r*   )Ú_version)r,   r6   s     r-   Ú__init__zArrowNamespace.__init__:   s   € ØˆŒˆˆr/   ú,)Ú	separatorÚsourcer#   r<   ÚstrÚkwdsr	   r   c               ó  — ddl m} |                     d¦  «        x}�7t          d|¦  «        j        |k    rd|› d|j        › d�}t          |¦  «        ‚n|                     |¬¦  «        |d<   t          |t          ¦  «        rXt          |                     d	¦  «        d
d¦  «        }t          |                     ¦   «                              |¦  «        ¦  «        }| j                              |j        |fi |¤Ž| ¬¦  «        S )Nr   )ÚcsvÚparse_optionszcsv.ParseOptionszD`separator` and `parse_options.delimiter` do not match: `separator`=z and `delimiter`=ú.)Ú	delimiterÚread_optionsÚencodingÚutf8©Úcontext)ÚpyarrowrA   Úgetr   rD   Ú	TypeErrorÚParseOptionsÚ
isinstancer   Úgetattrr   ÚreadÚencoder.   Úfrom_nativeÚread_csv)r,   r=   r<   r?   rA   rB   ÚmsgrF   s           r-   rS   zArrowNamespace.read_csv=   s-  € ð 	 ÐÐÐÐÐà!ŸXšX oÑ6Ô6Ð6ˆMÐCÝÐ&¨Ñ6Ô6Ô@ÀIÒMÐMðZØ#,ðZð ZØ?LÔ?VðZð Zð Zð õ   ‘n”nÐ$ð Nð %(×$4Ò$4¸yÐ$4Ñ$IÔ$IˆD�Ñ!õ �f�jÑ)Ô)ð 	=Ý˜tŸxšx¨Ñ7Ô7¸ÀVÑLÔLˆHÝ˜VŸ[š[™]œ]×1Ò1°(Ñ;Ô;Ñ<Ô<ˆFØŒ×*Ò*¨<¨3¬<¸Ð+GÐ+GÀ$Ð+GÐ+GÐQUÐ*ÑVÔVÐVr/   c                ó^   — ddl m} | j                              |j        |fi |¤Ž| ¬¦  «        S )Nr   )ÚparquetrH   )rJ   rV   r.   rR   Ú
read_table)r,   r=   r?   Úpqs       r-   Úread_parquetzArrowNamespace.read_parquetR   sD   € Ø)Ð)Ð)Ð)Ð)Ð)àŒ×*Ò*¨=¨2¬=¸Ð+HÐ+HÀ4Ð+HÐ+HÐRVÐ*ÑWÔWÐWr/   Úseriesr   ú%Iterator[ChunkedArrayAny | ScalarAny]c                ó   — d„ |D ¦   «         S )Nc              3  óL   K  — | ]}|j         r|j        d          n|j        V — Œ dS )r   N)Ú
_broadcastÚnative©Ú.0Úss     r-   ú	<genexpr>z0ArrowNamespace.extract_native.<locals>.<genexpr>Z   s5   è è € ÐJÐJ¸a˜qœ|Ð9�”˜”�°´ÐJÐJÐJÐJÐJÐJr/   © )r,   rZ   s     r-   Úextract_nativezArrowNamespace.extract_nativeW   s   € ð KÐJÀ6ÐJÑJÔJÐJr/   r   c                óH   ‡ — ‰                       ˆ fd„d„ d ‰ j        ¬¦  «        S )Nc                óZ   •— t          j        t          | j        ¦  «        gd‰¬¦  «        gS )NÚlen)ÚnamerI   )r   Úfrom_iterablerh   r_   )Údfr,   s    €r-   ú<lambda>z$ArrowNamespace.len.<locals>.<lambda>_   s+   ø€ ÝÔ)­3¨r¬y©>¬>Ð*:ÀÐPTÐUÑUÔUð€ r/   c                ó   — dgS )Nrh   rd   ©Ú_dfs    r-   rl   z$ArrowNamespace.len.<locals>.<lambda>b   s   € ¨u¨g€ r/   ©Úevaluate_output_namesÚalias_output_namesr6   ©r2   r9   r+   s   `r-   rh   zArrowNamespace.len\   sB   ø€ à�zŠzðð ð ð ð #6Ð"5Ø#Ø”Mð ñ 
ô 
ð 	
r/   Úvaluer$   ÚdtypeúIntoDType | Nonec                ó^   ‡ ‡‡‡— d	ˆˆ ˆfd„Š‰                       ˆfd„d„ d ‰ j        ¬¦  «        S )
NÚ_r   r'   r   c                óf   •— t          j        ‰gd‰¬¦  «        }‰r|                     ‰¦  «        S |S )NÚliteral©Údatari   rI   )r   rj   r   )rx   Úarrow_seriesru   r,   rt   s     €€€r-   Ú_lit_arrow_seriesz-ArrowNamespace.lit.<locals>._lit_arrow_seriesh   sG   ø€ Ý&Ô4Ø�W 9°dðñ ô ˆLð ð 0Ø#×(Ò(¨Ñ/Ô/Ð/ØÐr/   c                ó   •—  ‰| ¦  «        gS r*   rd   )rk   r~   s    €r-   rl   z$ArrowNamespace.lit.<locals>.<lambda>q   s   ø€ Ð)Ð)¨"Ñ-Ô-Ð.€ r/   c                ó   — dgS )Nrz   rd   rn   s    r-   rl   z$ArrowNamespace.lit.<locals>.<lambda>r   s   € ¨y¨k€ r/   rp   )rx   r   r'   r   rs   )r,   rt   ru   r~   s   ```@r-   ÚlitzArrowNamespace.litg   se   øøøø€ ð	 ð 	 ð 	 ð 	 ð 	 ð 	 ð 	 ð 	 ð �zŠzØ.Ð.Ð.Ð.Ø"9Ð"9Ø#Ø”Mð	 ñ 
ô 
ð 	
r/   ÚexprsÚignore_nullsÚboolc               ól   ‡‡— dˆˆfd„}| j                              |t          ‰Ž t          ‰Ž | ¬¦  «        S )Nrk   r   r'   úlist[ArrowSeries]c                ó˜   •‡ — t          j        ˆ fd„‰D ¦   «         ¦  «        }‰rd„ |D ¦   «         }t          t          j        |¦  «        gS )Nc              3  ó.   •K  — | ]} |‰¦  «        V — Œd S r*   rd   ©ra   Úerk   s     €r-   rc   z>ArrowNamespace.all_horizontal.<locals>.func.<locals>.<genexpr>y   ó+   øè è € Ð?UÐ?UÈ!ÀÀÀ"ÁÄÐ?UÐ?UÐ?UÐ?UÐ?UÐ?Ur/   c              3  óD   K  — | ]}|                      d dd¦  «        V — ŒdS )TN©Ú	fill_nullr`   s     r-   rc   z>ArrowNamespace.all_horizontal.<locals>.func.<locals>.<genexpr>{   s2   è è € ÐHÐH¸A˜!Ÿ+š+ d¨D°$Ñ7Ô7ÐHÐHÐHÐHÐHÐHr/   )r   rj   r   ÚoperatorÚand_©rk   rZ   r‚   rƒ   s   ` €€r-   Úfuncz+ArrowNamespace.all_horizontal.<locals>.funcx   sZ   øø€ Ý,1Ô,?Ð?UÐ?UÐ?UÐ?UÈuÐ?UÑ?UÔ?UÑ,UÔ,UˆFØð IØHÐHÀÐHÑHÔH�Ý�8œ=¨&Ñ1Ô1Ð2Ð2r/   ©r’   rq   rr   rI   ©rk   r   r'   r†   ©r2   Ú_from_callabler   r   ©r,   rƒ   r‚   r’   s    `` r-   Úall_horizontalzArrowNamespace.all_horizontalw   s^   øø€ ð	3ð 	3ð 	3ð 	3ð 	3ð 	3ð 	3ð Œz×(Ò(ØÝ"?ÀÐ"GÝ9¸5ÐAØð	 )ñ 
ô 
ð 	
r/   c               ól   ‡‡— dˆˆfd„}| j                              |t          ‰Ž t          ‰Ž | ¬¦  «        S )Nrk   r   r'   r†   c                ó˜   •‡ — t          j        ˆ fd„‰D ¦   «         ¦  «        }‰rd„ |D ¦   «         }t          t          j        |¦  «        gS )Nc              3  ó.   •K  — | ]} |‰¦  «        V — Œd S r*   rd   r‰   s     €r-   rc   z>ArrowNamespace.any_horizontal.<locals>.func.<locals>.<genexpr>‡   r‹   r/   c              3  óD   K  — | ]}|                      d dd¦  «        V — ŒdS )FNr�   r`   s     r-   rc   z>ArrowNamespace.any_horizontal.<locals>.func.<locals>.<genexpr>‰   s2   è è € ÐIÐI¸Q˜!Ÿ+š+ e¨T°4Ñ8Ô8ÐIÐIÐIÐIÐIÐIr/   )r   rj   r   r�   Úor_r‘   s   ` €€r-   r’   z+ArrowNamespace.any_horizontal.<locals>.func†   sZ   øø€ Ý,1Ô,?Ð?UÐ?UÐ?UÐ?UÈuÐ?UÑ?UÔ?UÑ,UÔ,UˆFØð JØIÐIÀ&ÐIÑIÔI�Ý�8œ<¨Ñ0Ô0Ð1Ð1r/   r“   r”   r•   r—   s    `` r-   Úany_horizontalzArrowNamespace.any_horizontal…   s^   øø€ ð	2ð 	2ð 	2ð 	2ð 	2ð 	2ð 	2ð Œz×(Ò(ØÝ"?ÀÐ"GÝ9¸5ÐAØð	 )ñ 
ô 
ð 	
r/   c                óh   ‡— dˆfd„}| j                              |t          ‰Ž t          ‰Ž | ¬¦  «        S )Nrk   r   r'   r†   c                ó”   •‡ — t          j        ˆ fd„‰D ¦   «         ¦  «        }d„ |D ¦   «         }t          t          j        |¦  «        gS )Nc              3  ó.   •K  — | ]} |‰¦  «        V — Œd S r*   rd   ©ra   Úexprrk   s     €r-   rc   z>ArrowNamespace.sum_horizontal.<locals>.func.<locals>.<genexpr>•   s+   øè è € Ð$@Ð$@°$ T T¨"¡X¤XÐ$@Ð$@Ð$@Ð$@Ð$@Ð$@r/   c              3  óF   K  — | ]}|                      d dd¬¦  «        V — ŒdS ©r   N)ÚstrategyÚlimitr�   r`   s     r-   rc   z>ArrowNamespace.sum_horizontal.<locals>.func.<locals>.<genexpr>–   s4   è è € ÐLÐLÀA�a—k’k !¨d¸$�kÑ?Ô?ÐLÐLÐLÐLÐLÐLr/   )r   rj   r   r�   Úadd)rk   ÚitrZ   r‚   s   `  €r-   r’   z+ArrowNamespace.sum_horizontal.<locals>.func”   sQ   øø€ ÝÔ$Ð$@Ð$@Ð$@Ð$@¸%Ð$@Ñ$@Ô$@Ñ@Ô@ˆBØLÐLÈÐLÑLÔLˆFÝ�8œ<¨Ñ0Ô0Ð1Ð1r/   r“   r”   r•   ©r,   r‚   r’   s    ` r-   Úsum_horizontalzArrowNamespace.sum_horizontal“   sX   ø€ ð	2ð 	2ð 	2ð 	2ð 	2ð 	2ð
 Œz×(Ò(ØÝ"?ÀÐ"GÝ9¸5ÐAØð	 )ñ 
ô 
ð 	
r/   c                ó¨   ‡‡— | j         j                             ¦   «         Šdˆˆfd„}| j                             |t          ‰Ž t          ‰Ž | ¬¦  «        S )Nrk   r   r'   r†   c                ó   •‡ — t          t          j        ˆ fd„‰D ¦   «         ¦  «        ¦  «        }d„ |D ¦   «         }ˆfd„|D ¦   «         }t          t          j        |¦  «        t          t          j        |¦  «        z  gS )Nc              3  ó.   •K  — | ]} |‰¦  «        V — Œd S r*   rd   r¢   s     €r-   rc   z?ArrowNamespace.mean_horizontal.<locals>.func.<locals>.<genexpr>¤   s+   øè è € Ð4PÐ4PÀ$°T°T¸"±X´XÐ4PÐ4PÐ4PÐ4PÐ4PÐ4Pr/   c                ó>   — g | ]}|                      d dd¬¦  «        ‘ŒS r¥   r�   r`   s     r-   ú
<listcomp>z@ArrowNamespace.mean_horizontal.<locals>.func.<locals>.<listcomp>¥   s*   € ÐVÐVÐVÀA�a—k’k !¨d¸$�kÑ?Ô?ÐVÐVÐVr/   c                ód   •— g | ],}d |                      ¦   «                              ‰¦  «        z
  ‘Œ-S )é   )Úis_nullr   )ra   rb   Úint_64s     €r-   r°   z@ArrowNamespace.mean_horizontal.<locals>.func.<locals>.<listcomp>¦   s4   ø€ ÐIÐIÐI°q�a˜!Ÿ)š)™+œ+×*Ò*¨6Ñ2Ô2Ñ2ÐIÐIÐIr/   )Útupler   rj   r   r�   r¨   )rk   Úexpr_resultsrZ   Únon_nar‚   r´   s   `   €€r-   r’   z,ArrowNamespace.mean_horizontal.<locals>.func£   sƒ   øø€ Ý ¥Ô!4Ð4PÐ4PÐ4PÐ4PÈ%Ð4PÑ4PÔ4PÑ!PÔ!PÑQÔQˆLØVÐVÈÐVÑVÔVˆFØIÐIÐIÐI¸LÐIÑIÔIˆFÝ�8œ<¨Ñ0Ô0µ6½(¼,ÈÑ3OÔ3OÑOÐPÐPr/   r“   r”   )r9   ÚdtypesÚInt64r2   r–   r   r   )r,   r‚   r’   r´   s    ` @r-   Úmean_horizontalzArrowNamespace.mean_horizontal    s{   øø€ Ø”Ô%×+Ò+Ñ-Ô-ˆð	Qð 	Qð 	Qð 	Qð 	Qð 	Qð 	Qð Œz×(Ò(ØÝ"?ÀÐ"GÝ9¸5ÐAØð	 )ñ 
ô 
ð 	
r/   c                ól   ‡ ‡— dˆˆ fd„}‰ j                              |t          ‰Ž t          ‰Ž ‰ ¬¦  «        S )Nrk   r   r'   r†   c                ó  •‡ — t          t          j        ˆ fd„‰D ¦   «         ¦  «        ¦  «        }t          t          j         ‰j        |Ž ¦  «        }t          t          |¦  «        |d         j	        ‰j
        ¬¦  «        gS )Nc              3  ó.   •K  — | ]} |‰¦  «        V — Œd S r*   rd   r¢   s     €r-   rc   z>ArrowNamespace.min_horizontal.<locals>.func.<locals>.<genexpr>²   ó+   øè è € Ð-IÐ-I¸4¨d¨d°2©h¬hÐ-IÐ-IÐ-IÐ-IÐ-IÐ-Ir/   r   ©ri   r6   )Úlistr   rj   r   ÚpcÚmin_element_wisere   r   r   ri   r9   ©rk   rZ   r_   r‚   r,   s   `  €€r-   r’   z+ArrowNamespace.min_horizontal.<locals>.func±   ó�   øø€ Ý�%Ô-Ð-IÐ-IÐ-IÐ-IÀ5Ð-IÑ-IÔ-IÑIÔIÑJÔJˆFÝ�BÔ/Ð1D°Ô1DÀfÐ1MÑNÔNˆFåÝ! &Ñ)Ô)°°q´	´ÈÌðñ ô ðð r/   r“   r”   r•   rª   s   `` r-   Úmin_horizontalzArrowNamespace.min_horizontal°   ó^   øø€ ð	ð 	ð 	ð 	ð 	ð 	ð 	ð Œz×(Ò(ØÝ"?ÀÐ"GÝ9¸5ÐAØð	 )ñ 
ô 
ð 	
r/   c                ól   ‡ ‡— dˆˆ fd„}‰ j                              |t          ‰Ž t          ‰Ž ‰ ¬¦  «        S )Nrk   r   r'   r†   c                ó  •‡ — t          t          j        ˆ fd„‰D ¦   «         ¦  «        ¦  «        }t          t          j         ‰j        |Ž ¦  «        }t          t          |¦  «        |d         j	        ‰j
        ¬¦  «        gS )Nc              3  ó.   •K  — | ]} |‰¦  «        V — Œd S r*   rd   r¢   s     €r-   rc   z>ArrowNamespace.max_horizontal.<locals>.func.<locals>.<genexpr>Ã   r¾   r/   r   r¿   )rÀ   r   rj   r   rÁ   Úmax_element_wisere   r   r   ri   r9   rÃ   s   `  €€r-   r’   z+ArrowNamespace.max_horizontal.<locals>.funcÂ   rÄ   r/   r“   r”   r•   rª   s   `` r-   Úmax_horizontalzArrowNamespace.max_horizontalÁ   rÆ   r/   ÚdfsúSequence[pa.Table]úpa.Tablec               óp   — | j         dk    rt          j        |d¬¦  «        S t          j        |d¬¦  «        S )N)é   Údefault)Úpromote_optionsT)Úpromote)Ú_backend_versionÚpaÚconcat_tables)r,   rÌ   s     r-   Ú_concat_diagonalzArrowNamespace._concat_diagonalÒ   s<   € ØÔ  EÒ)Ð)ÝÔ# C¸ÐCÑCÔCÐCÝÔ ¨TÐ2Ñ2Ô2Ð2r/   c               óð   — t          t          j        d„ |D ¦   «         ¦  «        ¦  «        }t          t          j        d„ |D ¦   «         ¦  «        ¦  «        }t          j                             ||¬¦  «        S )Nc              3  ó$   K  — | ]}|j         V — Œd S r*   )Úcolumn_names©ra   rk   s     r-   rc   z4ArrowNamespace._concat_horizontal.<locals>.<genexpr>Ø   s$   è è € Ð(GÐ(G¸R¨¬Ð(GÐ(GÐ(GÐ(GÐ(GÐ(Gr/   c              3  ó>   K  — | ]}|                      ¦   «         V — Œd S r*   )ÚitercolumnsrÛ   s     r-   rc   z4ArrowNamespace._concat_horizontal.<locals>.<genexpr>Ù   s,   è è € Ð*JÐ*JÀ¨2¯>ª>Ñ+;Ô+;Ð*JÐ*JÐ*JÐ*JÐ*JÐ*Jr/   )Únames)rÀ   r   rj   rµ   rÕ   ÚTableÚfrom_arrays)r,   rÌ   rÞ   Úarrayss       r-   Ú_concat_horizontalz!ArrowNamespace._concat_horizontal×   sk   € Ý•UÔ(Ð(GÐ(GÀ3Ð(GÑ(GÔ(GÑGÔGÑHÔHˆÝ•uÔ*Ð*JÐ*JÀcÐ*JÑ*JÔ*JÑJÔJÑKÔKˆÝŒx×#Ò# F°%Ð#Ñ8Ô8Ð8r/   c               óÒ   — |d         j         }t          |dd …         d¬¦  «        D ]-\  }}|j         x}|k    rd|› d|› d|› d�}t          |¦  «        ‚Œ.t          j        |¦  «        S )Nr   r²   )Ústartz>unable to vstack, column names don't match:
   - dataframe 0: z
   - dataframe z: ú
)rÚ   Ú	enumeraterL   rÕ   rÖ   )r,   rÌ   Úcols_0Úirk   Úcols_currentrT   s          r-   Ú_concat_verticalzArrowNamespace._concat_verticalÜ   s¤   € Ø�Q”Ô$ˆÝ˜s 1 2 2œw¨aÐ0Ñ0Ô0ð 	%ð 	%‰EˆAˆrØ "¤Ð/�°FÒ:Ð:ð<Ø)/ð<ð <à&'ð<ð <à+7ð<ð <ð <ð õ
   ‘n”nÐ$ð ;õ Ô Ñ$Ô$Ð$r/   r   c                ó*   — t          j        | ¦  «        S r*   )r   Úfrom_namespacer+   s    r-   Ú	selectorszArrowNamespace.selectorsè   s   € å%Ô4°TÑ:Ô:Ð:r/   c               ót   ‡ ‡‡‡— dˆˆˆ ˆfd„}‰ j                              |t          ‰Ž t          ‰Ž ‰ ¬¦  «        S )Nrk   r   r'   r†   c                ó°  •‡ — t          t          j        ˆ fd„‰D ¦   «         ¦  «        ¦  «        }|d         j        }‰rdnd}t	           ‰j        |Ž d‰iŽ\  }}|^}}‰r@t          j        }t          |d„ |g|¢R D ¦   «         ¦  «        }	t          j	        |	d|¦  «        }t          j
        }
‰                      |
|g|¢|‘R d|iŽ|‰j        ¬	¦  «        }|gS )
Nc              3  ó.   •K  — | ]} |‰¦  «        V — Œd S r*   rd   r¢   s     €r-   rc   z:ArrowNamespace.concat_str.<locals>.func.<locals>.<genexpr>ð   r¾   r/   r   ÚskipÚ	emit_nullr<   c              3  ó>   K  — | ]}t          j        |¦  «        V — Œd S r*   )rÁ   r³   )ra   Úcs     r-   rc   z:ArrowNamespace.concat_str.<locals>.func.<locals>.<genexpr>ý   s*   è è € Ð(OÐ(O¸1­¬°A©¬Ð(OÐ(OÐ(OÐ(OÐ(OÐ(Or/   Ú Únull_handlingr¿   )rÀ   r   rj   ri   r   re   rÁ   r�   r   Úif_elseÚbinary_join_element_wiser5   r9   )rk   rZ   ri   rö   r©   Úseparator_scalarÚfirstÚrestr�   Úall_nullÚ
concat_strÚ	compliantr‚   rƒ   r,   r<   s   `           €€€€r-   r’   z'ArrowNamespace.concat_str.<locals>.funcï   s&  øø€ Ý�%Ô-Ð-IÐ-IÐ-IÐ-IÀ5Ð-IÑ-IÔ-IÑIÔIÑJÔJˆFØ˜!”9”>ˆDà&Ð7��¨Kð õ $CØ$�Ô$ fÐ-ð$Ø9Bð$ð $Ñ ˆBÐ ð ˆLˆE�DØð 8õ $&¤7�Ý! $Ð(OÐ(OÀÀÈÀÀÐ(OÑ(OÔ(OÑPÔP�Ýœ
 8¨R°Ñ7Ô7�õ &(Ô%@ˆJØŸšØ�
˜5ÐW 4ÐWÐ)9ÐWÐWÐWÈÐWÐWØØœð %ñ ô ˆIð
 �;Ðr/   r“   r”   r•   )r,   r<   rƒ   r‚   r’   s   ```` r-   rý   zArrowNamespace.concat_strì   sj   øøøø€ ð	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð4 Œz×(Ò(ØÝ"?ÀÐ"GÝ9¸5ÐAØð	 )ñ 
ô 
ð 	
r/   c                ól   ‡ ‡— dˆˆ fd„}‰ j                              |t          ‰Ž t          ‰Ž ‰ ¬¦  «        S )Nrk   r   r'   r†   c                óà   •‡ — ‰j         j        } |t          j        ˆ fd„‰D ¦   «         ¦  «        Ž ^}}t	          t          j        |j        gd„ |D ¦   «         ¢R Ž |j        ‰j	        ¬¦  «        gS )Nc              3  ó.   •K  — | ]} |‰¦  «        V — Œd S r*   rd   r¢   s     €r-   rc   z8ArrowNamespace.coalesce.<locals>.func.<locals>.<genexpr>  s+   øè è € Ð=YÐ=YÈ4¸d¸dÀ2¹h¼hÐ=YÐ=YÐ=YÐ=YÐ=YÐ=Yr/   c              3  ó$   K  — | ]}|j         V — Œd S r*   )r_   r`   s     r-   rc   z8ArrowNamespace.coalesce.<locals>.func.<locals>.<genexpr>  s$   è è € Ð5OÐ5OÀ1°a´hÐ5OÐ5OÐ5OÐ5OÐ5OÐ5Or/   r¿   )
r5   Ú_align_full_broadcastr   rj   r   rÁ   Úcoalescer_   ri   r9   )rk   ÚalignÚinit_seriesrZ   r‚   r,   s   `   €€r-   r’   z%ArrowNamespace.coalesce.<locals>.func  sŽ   øø€ Ø”LÔ6ˆEØ#( 5­%Ô*=Ð=YÐ=YÐ=YÐ=YÐSXÐ=YÑ=YÔ=YÑ*YÔ*YÐ#ZÐ ˆK˜&åÝ”K Ô 2ÐPÐ5OÐ5OÈÐ5OÑ5OÔ5OÐPÐPÐPØ$Ô)Ø œMðñ ô ðð r/   r“   r”   r•   rª   s   `` r-   r  zArrowNamespace.coalesce  s^   øø€ ð		ð 		ð 		ð 		ð 		ð 		ð 		ð Œz×(Ò(ØÝ"?ÀÐ"GÝ9¸5ÐAØð	 )ñ 
ô 
ð 	
r/   c                ól   ‡ ‡— dˆˆ fd„}‰ j                              |t          ‰Ž t          ‰Ž ‰ ¬¦  «        S )Nrk   r   r'   r†   c                ój  •‡ — t          t          j        ˆ fd„‰D ¦   «         ¦  «        ¦  «        }t          d„ |D ¦   «         ¦  «         |d         j        }t          j        d„ |D ¦   «         dt          d„ |D ¦   «         ¦  «        iŽ}t          j        |g¦  «        }t          ||‰j
        ¬¦  «        gS )Nc              3  ó.   •K  — | ]} |‰¦  «        V — Œd S r*   rd   r¢   s     €r-   rc   z6ArrowNamespace.struct.<locals>.func.<locals>.<genexpr>%  ó+   øè è € Ð.JÐ.J¸D¨t¨t°B©x¬xÐ.JÐ.JÐ.JÐ.JÐ.JÐ.Jr/   c                ó   — g | ]	}|j         ‘Œ
S rd   ©ri   r`   s     r-   r°   z7ArrowNamespace.struct.<locals>.func.<locals>.<listcomp>&  s   € Ð*BÐ*BÐ*B°a¨1¬6Ð*BÐ*BÐ*Br/   r   c              3  ód   K  — | ]+}t          |¦  «        d k    r|j        n|j        d         V — Œ,dS )r²   r   N)rh   r_   r`   s     r-   rc   z6ArrowNamespace.struct.<locals>.func.<locals>.<genexpr>*  s<   è è € ÐJÐJ¸a�c !™fœf qšj˜j�!”(�(¨a¬h°q¬kÐJÐJÐJÐJÐJÐJr/   Úfield_namesc              3  ó$   K  — | ]}|j         V — Œd S r*   r  r`   s     r-   rc   z6ArrowNamespace.struct.<locals>.func.<locals>.<genexpr>+  s$   è è € Ð!9Ð!9¨Q !¤&Ð!9Ð!9Ð!9Ð!9Ð!9Ð!9r/   r¿   )rµ   r   rj   r   ri   rÁ   Úmake_structrÕ   r   r   r9   )rk   rZ   ri   Ústruct_arrayÚresultr‚   r,   s   `    €€r-   r’   z#ArrowNamespace.struct.<locals>.func$  sÈ   øø€ Ý�5Ô.Ð.JÐ.JÐ.JÐ.JÀEÐ.JÑ.JÔ.JÑJÔJÑKÔKˆFÝ)Ð*BÐ*B¸6Ð*BÑ*BÔ*BÑCÔCÐCØ˜!”9”>ˆDåœ>ØJÐJÀ6ÐJÑJÔJðå!Ð!9Ð!9°&Ð!9Ñ!9Ô!9Ñ9Ô9ðð ˆLõ Ô% | nÑ5Ô5ˆFÝ ¨T¸4¼=ÐIÑIÔIÐJÐJr/   r“   r”   r•   rª   s   `` r-   ÚstructzArrowNamespace.struct#  se   øø€ ð
	Kð 
	Kð 
	Kð 
	Kð 
	Kð 
	Kð 
	Kð Œz×(Ò(ØÝ"?ÀÐ"GÝ9¸5ÐAØð	 )ñ 
ô 
ð 	
r/   c                ól   ‡ ‡— dˆˆ fd„}‰ j                              |t          ‰Ž t          ‰Ž ‰ ¬¦  «        S )Nrk   r   r'   r†   c                óØ   •‡ — t          t          j        ˆ fd„‰D ¦   «         ¦  «        ¦  «        }t          d„ |D ¦   «         ¦  «        }t	          ||d         j        ‰j        ¬¦  «        gS )Nc              3  ó.   •K  — | ]} |‰¦  «        V — Œd S r*   rd   r¢   s     €r-   rc   z4ArrowNamespace.list.<locals>.func.<locals>.<genexpr>9  r
  r/   c                ó@   — g | ]}|j                              ¦   «         ‘ŒS rd   )r_   Úcombine_chunksr`   s     r-   r°   z5ArrowNamespace.list.<locals>.func.<locals>.<listcomp>;  s&   € Ð&QÐ&QÐ&QÀQ q¤x×'>Ò'>Ñ'@Ô'@Ð&QÐ&QÐ&Qr/   r   r¿   )rµ   r   rj   r   r   ri   r9   )rk   rZ   r  r‚   r,   s   `  €€r-   r’   z!ArrowNamespace.list.<locals>.func8  sm   øø€ Ý�5Ô.Ð.JÐ.JÐ.JÐ.JÀEÐ.JÑ.JÔ.JÑJÔJÑKÔKˆFå%Ð&QÐ&QÈ&Ð&QÑ&QÔ&QÑRÔRˆFÝ ¨V°A¬Y¬^ÀTÄ]ÐSÑSÔSÐTÐTr/   r“   r”   r•   rª   s   `` r-   rÀ   zArrowNamespace.list7  se   øø€ ð	Uð 	Uð 	Uð 	Uð 	Uð 	Uð 	Uð Œz×(Ò(ØÝ"?ÀÐ"GÝ9¸5ÐAØð	 )ñ 
ô 
ð 	
r/   NÚwhenr   ÚthenÚ	otherwiseúChunkedArrayAny | Nonec                óÎ   — |€'t          j        t          |¦  «        |j        ¦  «        n|}t	          j        |                     t          j        d¦  «        ¦  «        ||¦  «        S )NF)rÕ   Únullsrh   ÚtyperÁ   r÷   rŽ   Úscalar)r,   r  r  r  s       r-   Ú_if_then_elsezArrowNamespace._if_then_elseE  sR   € ð 7@Ð6G•B”H�S ™YœY¨¬	Ñ2Ô2Ð2ÈYˆ	ÝŒz˜$Ÿ.š.­¬°5Ñ)9Ô)9Ñ:Ô:¸DÀ)ÑLÔLÐLr/   ÚaÚbÚmethodr!   c               óº   ‡ ‡‡— |dk    rd}t          |¦  «        ‚d	ˆˆˆ fd„}‰ j                             |t          ‰‰¦  «        t	          ‰‰¦  «        ‰ ¬¦  «        S )
NÚpearsonz4Only 'pearson' correlation is supported for Pyarrow.rk   r   r'   r†   c                óü  •— |                       ‰¦  «        }|j        }|                       ‰¦  «        j        }t          j        t          j        |¦  «        t          j        |¦  «        ¦  «        }t          j        ||¦  «        }t          j        ||¦  «        }t          j        |t          j        |¦  «        ¦  «        }t          j        |t          j        |¦  «        ¦  «        }t          j        t          j        ||¦  «        ¦  «        }t          j	        |¦  «        }t          j	        |¦  «        }	t          j
        |t          j        ||	¦  «        ¦  «        }
t          j        |
g|j        ‰¬¦  «        gS )Nr{   )Ú_evaluate_single_output_exprr_   rÁ   r�   Úis_validÚfilterÚsubtractÚmeanÚmultiplyÚstddevÚdivider   rj   ri   )rk   Úa_seriesÚarr1Úarr2ÚvalidÚdev1Údev2Ú
covarianceÚstd1Ústd2Úcorrelationr"  r#  r,   s              €€€r-   r’   z!ArrowNamespace.corr.<locals>.funcS  s(  ø€ Ø×6Ò6°qÑ9Ô9ˆHØ”?ˆDØ×2Ò2°1Ñ5Ô5Ô<ˆDÝ”G�BœK¨Ñ-Ô-­r¬{¸4Ñ/@Ô/@ÑAÔAˆEÝ”9˜T 5Ñ)Ô)ˆDÝ”9˜T 5Ñ)Ô)ˆDå”;˜t¥R¤W¨T¡]¤]Ñ3Ô3ˆDÝ”;˜t¥R¤W¨T¡]¤]Ñ3Ô3ˆDåœ¥¤¨T°4Ñ!8Ô!8Ñ9Ô9ˆJå”9˜T‘?”?ˆDÝ”9˜T‘?”?ˆDåœ) Jµ´¸DÀ$Ñ0GÔ0GÑHÔHˆKåÔ)Ø%˜¨X¬]ÀDðñ ô ðð r/   r“   r”   )ÚNotImplementedErrorr2   r–   r   r   )r,   r"  r#  r$  rT   r’   s   ```   r-   ÚcorrzArrowNamespace.corrN  sŒ   øøø€ Ø�YÒÐØHˆCÝ% cÑ*Ô*Ð*ð	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð. Œz×(Ò(ØÝ"?ÀÀ1Ñ"EÔ"EÝ9¸!¸QÑ?Ô?Øð	 )ñ 
ô 
ð 	
r/   ÚddofÚintc               ó�   ‡ ‡‡‡— dˆˆˆˆ fd„}‰ j                              |t          ‰‰¦  «        t          ‰‰¦  «        ‰ ¬¦  «        S )Nrk   r   r'   r†   c                ó0  •— |                       ‰
¦  «        }|j        }|                       ‰¦  «        j        }t          j        t          j        |¦  «        t          j        |¦  «        ¦  «        }t          j        ||¦  «        }t          j        ||¦  «        }t          |¦  «        }|‰z
  dk    r(t          j        d t          j	        ¦   «         ¬¦  «        }nŸt          j
        |t          j        |¦  «        ¦  «        }t          j
        |t          j        |¦  «        ¦  «        }t          j        t          j        ||¦  «        ¦  «        }	t          j        |	t          j        |‰z
  ¦  «        ¦  «        }t          j        |g|j        ‰¬¦  «        gS )Nr   )r  r{   )r(  r_   rÁ   r�   r)  r*  rh   rÕ   r   Úfloat64r+  r,  Úsumr-  r/  r   rj   ri   )rk   r0  r1  r2  r3  Únr6  r4  r5  Ú	numeratorr"  r#  r<  r,   s             €€€€r-   r’   z ArrowNamespace.cov.<locals>.funcr  sF  ø€ Ø×6Ò6°qÑ9Ô9ˆHØ”?ˆDØ×2Ò2°1Ñ5Ô5Ô<ˆDÝ”G�BœK¨Ñ-Ô-­r¬{¸4Ñ/@Ô/@ÑAÔAˆEÝ”9˜T 5Ñ)Ô)ˆDÝ”9˜T 5Ñ)Ô)ˆDÝ�D‘	”	ˆAØ�4‰x˜1Š}ˆ}õ  œY tµ"´*±,´,Ð?Ñ?Ô?�
�
å”{ 4­¬°©¬Ñ7Ô7�Ý”{ 4­¬°©¬Ñ7Ô7�ÝœF¥2¤;¨t°TÑ#:Ô#:Ñ;Ô;�	ÝœY yµ"´)¸AÀ¹HÑ2EÔ2EÑFÔF�
åÔ)Ø$˜¨H¬MÀ4ðñ ô ðð r/   r“   r”   r•   )r,   r"  r#  r<  r’   s   ```` r-   ÚcovzArrowNamespace.covq  st   øøøø€ ð	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð. Œz×(Ò(ØÝ"?ÀÀ1Ñ"EÔ"EÝ9¸!¸QÑ?Ô?Øð	 )ñ 
ô 
ð 	
r/   )r'   r(   )r'   r0   )r'   r3   )r6   r    r'   r7   )r=   r#   r<   r>   r?   r	   r'   r   )r=   r#   r?   r	   r'   r   )rZ   r   r'   r[   )r'   r   )rt   r$   ru   rv   r'   r   )r‚   r   rƒ   r„   r'   r   )r‚   r   r'   r   )rÌ   rÍ   r'   rÎ   )r'   r   )r‚   r   r<   r>   rƒ   r„   r'   r   r*   )r  r   r  r   r  r  r'   r   )r"  r   r#  r   r$  r!   r'   r   )r"  r   r#  r   r<  r=  r'   r   )!Ú__name__Ú
__module__Ú__qualname__r   ÚPYARROWÚ_implementationÚpropertyr.   r2   r5   r:   rS   rY   re   rh   r�   r˜   rž   r«   rº   rÅ   rË   r×   râ   rê   rí   rý   r  r  rÀ   r!  r;  rD  rd   r/   r-   r&   r&   )   sƒ  € € € € € ð %Ô,€Oàðð ð ñ „Xðð ðð ð ñ „Xðð ðð ð ñ „Xðð ð  ð  ð  ð =@ðWð Wð Wð Wð Wð Wð*Xð Xð Xð Xð
Kð Kð Kð Kð
	
ð 	
ð 	
ð 	
ð
ð 
ð 
ð 
ð 
ð 
ð 
ð 
ð
ð 
ð 
ð 
ð
ð 
ð 
ð 
ð
ð 
ð 
ð 
ð 
ð 
ð 
ð 
ð"
ð 
ð 
ð 
ð"3ð 3ð 3ð 3ð
9ð 9ð 9ð 9ð

%ð 
%ð 
%ð 
%ð ð;ð ;ð ;ñ „Xð;ð"
ð "
ð "
ð "
ðH
ð 
ð 
ð 
ð&
ð 
ð 
ð 
ð(
ð 
ð 
ð 
ð$ -1ð	Mð Mð Mð Mð Mð!
ð !
ð !
ð !
ðF
ð 
ð 
ð 
ð 
ð 
r/   r&   r   )7Ú
__future__r   r�   Ú	functoolsr   Úior   r   Ú	itertoolsr   Útypingr   r	   r
   r   rJ   rÕ   Úpyarrow.computeÚcomputerÁ   Únarwhals._arrow.dataframer   Únarwhals._arrow.exprr   Únarwhals._arrow.selectorsr   Únarwhals._arrow.seriesr   Únarwhals._arrow.utilsr   r   r   Únarwhals._compliantr   Únarwhals._expression_parsingr   r   Únarwhals._utilsr   r   Úcollections.abcr   r   Únarwhals._arrow.typingr   r   r   r    Únarwhals.typingr!   r"   r#   r$   rß   r&   rd   r/   r-   ú<module>r]     sM  ðØ "Ð "Ð "Ð "Ð "Ð "à €€€Ø Ð Ð Ð Ð Ð Ø "Ð "Ð "Ð "Ð "Ð "Ð "Ð "Ø Ð Ð Ð Ð Ð Ø 4Ð 4Ð 4Ð 4Ð 4Ð 4Ð 4Ð 4Ð 4Ð 4Ð 4Ð 4à Ð Ð Ð Ø Ð Ð Ð Ð Ð à 4Ð 4Ð 4Ð 4Ð 4Ð 4Ø *Ð *Ð *Ð *Ð *Ð *Ø <Ð <Ð <Ð <Ð <Ð <Ø .Ð .Ð .Ð .Ð .Ð .ðð ð ð ð ð ð ð ð ð ð
 /Ð .Ð .Ð .Ð .Ð .ðð ð ð ð ð ð ð ð JÐ IÐ IÐ IÐ IÐ IÐ IÐ Iàð 
Ø2Ð2Ð2Ð2Ð2Ð2Ð2Ð2àMÐMÐMÐMÐMÐMÐMÐMÐMÐMØ'Ð'Ð'Ð'Ð'Ð'ðð ð ð ð ð ð ð ð ð ð ð ðe
ð e
ð e
ð e
ð e
Ø�> ;°	¸2¼8ÐEVÐVÔWñe
ô e
ð e
ð e
ð e
r/   