
    .i+I              
         U d dl mZ d dlZd dlZd dlmZ d dlmZ d dlm	Z	m
Z
mZmZmZ d dlmZ d dlmZ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 d dlmZ d dl m!Z! e	r d dl"m#Z#m$Z$ d dl%m&Z& d dl m'Z'm(Z( d dl)m*Z*m+Z+m,Z, e
Z-de.d<   	 ed    Z/de.d<   ed   Z0de.d<   ee/e0f   Z1de.d<   d Z2de.d<   dZ3de.d<    G d deeeeeef         Z4 G d deeef         Z5y)    )annotationsNreduce)chain)TYPE_CHECKINGAnyLiteralProtocoloverload)EagerNamespace)combine_alias_output_namescombine_evaluate_output_namesPandasLikeDataFramePandasLikeExpr)PandasSelectorNamespacePandasLikeSeries)NativeDataFrameTNativeSeriesT)is_non_nullable_boolean)
zip_strict)IterableSequence)	TypeAlias)ImplementationVersion)CorrelationMethod	IntoDTypePythonLiteralr   
Incomplete	_Vertical   _HorizontalAxisVERTICAL
HORIZONTALc                  @   e Zd Zedd       Zedd       Zedd       Zedd       ZddZddZ	ddZ
ddZdd	Z	 	 	 	 	 	 d d
Z	 	 	 	 	 	 d dZddZddZddZed!d       Zd"dZ	 	 	 	 d#dZd"dZ	 	 	 	 	 	 	 	 d$dZddZ	 d%	 	 	 	 	 	 	 d&dZ	 	 	 	 	 	 	 	 d'dZy)(PandasLikeNamespacec                    t         S Nr   selfs    \/opt/house_gps_flutter/.venv/lib/python3.12/site-packages/narwhals/_pandas_like/namespace.py
_dataframezPandasLikeNamespace._dataframe4   s    ""    c                    t         S r,   r   r-   s    r/   _exprzPandasLikeNamespace._expr8   s    r1   c                    t         S r,   r   r-   s    r/   _serieszPandasLikeNamespace._series<   s    r1   c                ,    t        j                  |       S r,   )r   from_namespacer-   s    r/   	selectorszPandasLikeNamespace.selectors@   s    &55d;;r1   c                     || _         || _        y r,   )_implementation_version)r.   implementationversions      r/   __init__zPandasLikeNamespace.__init__D   s    -r1   c                f    dfd}| j                   j                  |t         t         |       S )Nc                8      fdD        }t        d |      gS )Nc              3  <   K   | ]  } |      D ]  }|   y wr,    ).0r3   sdfs      r/   	<genexpr>z=PandasLikeNamespace.coalesce.<locals>.func.<locals>.<genexpr>J   s!     >EE"I>qa>a>s   c                *    | j                  |d d       S )Nstrategylimit	fill_null)xys     r/   <lambda>z<PandasLikeNamespace.coalesce.<locals>.func.<locals>.<lambda>L   s    AKKDK$M r1   r   )rE   seriesexprss   ` r/   funcz*PandasLikeNamespace.coalesce.<locals>.funcI   s"    >U>FMvV r1   rR   evaluate_output_namesalias_output_namescontextrE   r   returnzlist[PandasLikeSeries]r3   _from_callabler   r   r.   rQ   rR   s    ` r/   coalescezPandasLikeNamespace.coalesceH   s<    	 zz(("?"G95A	 ) 
 	
r1   c                j     d fdt        fdd d  j                   j                        S )Nc                   t        t        t        t        f      r	 dd l}dd l}ddlm} t        t              rt              n}r |j                        nd }|j                  |g|      }j                  j                         }	|	j                  |j                  j                  |      d| j                   j"                  dd       }
j$                  j'                  |
      S j$                  j)                  gd| j                   j"                  dd 	      }r|j+                        S |S # t        $ r}d}t        |      |d }~ww xY w)
Nr   zqNested structures require pyarrow to be installed for pandas backend. Please install pyarrow: pip install pyarrow)narwhals_to_native_dtype)typeliteralr$   nameindex)rV   )datarc   rd   rV   )
isinstancelisttupledictpandaspyarrowImportErrornarwhals._arrow.utilsr_   r;   arrayr:   to_native_namespaceSeriesarraysArrowExtensionArray_native_framerd   r5   from_nativefrom_iterablecast)rE   pdpaexcmsg_to_arrow_dtypearray_valuepa_dtypepa_arraynspandas_series_nativepandas_like_seriesdtyper.   values               r/   _lit_pandas_seriesz3PandasLikeNamespace.lit.<locals>._lit_pandas_seriesW   sX   %$t!454'( .8u-Ed5k5DI?5$--@t88[M8A ))==?')yyII11(;"**0015 (1 ($ ||//0Dd/SS!%!;!;W&&,,Qq1	 "< " )..u55%%C # 4F  &c*34s   D; ;	EEEc                     |       gS r,   rB   )rE   r   s    r/   rO   z)PandasLikeNamespace.lit.<locals>.<lambda>   s    *2./ r1   c                    dgS )Nra   rB   _dfs    r/   rO   z)PandasLikeNamespace.lit.<locals>.<lambda>   s    yk r1   rT   rU   r<   r=   )rE   r   rX   r   r   r:   r;   )r.   r   r   r   s   ```@r/   litzPandasLikeNamespace.litV   s1    &	&P /"9#//MM
 	
r1   c                T     t         fdd d  j                   j                        S )Nc                n    j                   j                  t        | j                        gddg      gS )Nlenr   )rc   rd   rV   )r5   ru   r   rs   )rE   r.   s    r/   rO   z)PandasLikeNamespace.len.<locals>.<lambda>   s:    **))*+%sD +  r1   c                    dgS )Nr   rB   r   s    r/   rO   z)PandasLikeNamespace.len.<locals>.<lambda>   s    ug r1   r   r   r-   s   `r/   r   zPandasLikeNamespace.len   s-    
 #6#//MM

 
	
r1   c                f    dfd}| j                   j                  |t         t         |       S )Nc                     t        j                   fdD              }d |D        }t        t        j                  |      gS )Nc              3  .   K   | ]  } |        y wr,   rB   rC   exprrE   s     r/   rF   zCPandasLikeNamespace.sum_horizontal.<locals>.func.<locals>.<genexpr>   s     $@$T"X$@   c              3  B   K   | ]  }|j                  d dd        yw)r   NrK   rC   rD   s     r/   rF   zCPandasLikeNamespace.sum_horizontal.<locals>.func.<locals>.<genexpr>   s     DAQ[[D$7D   )r   ru   r   operatoradd)rE   itnative_seriesrQ   s   `  r/   rR   z0PandasLikeNamespace.sum_horizontal.<locals>.func   s7    $$$@%$@@BDDM8<<788r1   rS   rW   rY   r[   s    ` r/   sum_horizontalz"PandasLikeNamespace.sum_horizontal   s<    	9
 zz(("?"G95A	 ) 
 	
r1   c               j    dfd}| j                   j                  |t         t         |       S )Nc                    D cg c]  } ||       D ]  }|  }}}st        d |D              rd}t        |      r	d |D        n
t        |      }t        t        j
                  |      gS c c}}w )Nc              3     K   | ]=  }|j                   j                  d k(  xr |j                         j                          ? ywobjectNnativer   is_nullanyr   s     r/   rF   zCPandasLikeNamespace.all_horizontal.<locals>.func.<locals>.<genexpr>   7      (EF(*@qyy{/@@(   AAzwCannot use `ignore_nulls=False` in `all_horizontal` for non-nullable NumPy-backed pandas Series when nulls are present.c              3  \   K   | ]$  }t        |      r|n|j                  d dd       & yw)TNr   rL   r   s     r/   rF   zCPandasLikeNamespace.all_horizontal.<locals>.func.<locals>.<genexpr>   s4        13AT4QU9VV   *,)r   
ValueErroriterr   r   and_rE   r3   rD   rP   rz   r   rQ   ignore_nullss         r/   rR   z0PandasLikeNamespace.all_horizontal.<locals>.func   s    %*>EE"I>qa>a>F>C (JP( %
 P o%   $ &\  8=="-..# ?   A2rS   rW   rY   r.   r   rQ   rR   s    `` r/   all_horizontalz"PandasLikeNamespace.all_horizontal   s<    	/( zz(("?"G95A	 ) 
 	
r1   c               j    dfd}| j                   j                  |t         t         |       S )Nc                    D cg c]  } ||       D ]  }|  }}}st        d |D              rd}t        |      r	d |D        n
t        |      }t        t        j
                  |      gS c c}}w )Nc              3     K   | ]=  }|j                   j                  d k(  xr |j                         j                          ? ywr   r   r   s     r/   rF   zCPandasLikeNamespace.any_horizontal.<locals>.func.<locals>.<genexpr>   r   r   zwCannot use `ignore_nulls=False` in `any_horizontal` for non-nullable NumPy-backed pandas Series when nulls are present.c              3  \   K   | ]$  }t        |      r|n|j                  d dd       & yw)FNr   r   s     r/   rF   zCPandasLikeNamespace.any_horizontal.<locals>.func.<locals>.<genexpr>   s4        13AUDRV9WWr   )r   r   r   r   r   or_r   s         r/   rR   z0PandasLikeNamespace.any_horizontal.<locals>.func   s    %*>EE"I>qa>a>F>C (JP( %
 P o%   $ &\  8<<,--# ?r   rS   rW   rY   r   s    `` r/   any_horizontalz"PandasLikeNamespace.any_horizontal   s<    	.( zz(("?"G95A	 ) 
 	
r1   c                f    dfd}| j                   j                  |t         t         |       S )Nc                    D cg c]  } ||       D ]  }|  }}}d |D        }d |D        }t        t        j                  |      t        t        j                  |      z  gS c c}}w )Nc              3  D   K   | ]  }|j                  d dd        yw)r   NrH   rK   r   s     r/   rF   zDPandasLikeNamespace.mean_horizontal.<locals>.func.<locals>.<genexpr>   s     VAakk!d$k?Vs    c              3  B   K   | ]  }d |j                         z
    yw)r$   N)r   r   s     r/   rF   zDPandasLikeNamespace.mean_horizontal.<locals>.func.<locals>.<genexpr>   s     <!a!))+o<r   )r   r   r   )rE   r3   rD   expr_resultsrP   non_narQ   s         r/   rR   z1PandasLikeNamespace.mean_horizontal.<locals>.func   sd    +0D%%)DQADADLDVVF<|<F8<<06(,,3OOPP Es   A(rS   rW   rY   r[   s    ` r/   mean_horizontalz#PandasLikeNamespace.mean_horizontal   s=    	Q zz(("?"G95A	 ) 
 	
r1   c                j     d fd} j                   j                  |t         t                S )Nc                6    t        t        j                   fdD                    }t        j	                  d |D        d      j
                  j                  d      j                  j                        j                  |d   j                        gS )	Nc              3  .   K   | ]  } |        y wr,   rB   r   s     r/   rF   zCPandasLikeNamespace.min_horizontal.<locals>.func.<locals>.<genexpr>        -I4d2h-Ir   c              3  <   K   | ]  }|j                           y wr,   to_framer   s     r/   rF   zCPandasLikeNamespace.min_horizontal.<locals>.func.<locals>.<genexpr>        6!6   
horizontalhowr$   axisr<   r=   r   )rg   r   ru   r   concatrs   minr:   r;   aliasrc   rE   rP   rQ   r.   s   ` r/   rR   z0PandasLikeNamespace.min_horizontal.<locals>.func   s~    %---I5-IIJF KK6v6L   #mCCQCK#'#7#7 MM %q	' r1   rS   rW   rY   r[   s   `` r/   min_horizontalz"PandasLikeNamespace.min_horizontal   <    
	 zz(("?"G95A	 ) 
 	
r1   c                j     d fd} j                   j                  |t         t                S )Nc                6    t        t        j                   fdD                    }t        j	                  d |D        d      j
                  j                  d      j                  j                        j                  |d   j                        gS )	Nc              3  .   K   | ]  } |        y wr,   rB   r   s     r/   rF   zCPandasLikeNamespace.max_horizontal.<locals>.func.<locals>.<genexpr>  r   r   c              3  <   K   | ]  }|j                           y wr,   r   r   s     r/   rF   zCPandasLikeNamespace.max_horizontal.<locals>.func.<locals>.<genexpr>  r   r   r   r   r$   r   r   r   )rg   r   ru   r   r   r   maxr:   r;   r   rc   r   s   ` r/   rR   z0PandasLikeNamespace.max_horizontal.<locals>.func  s~    %---I5-IIJF KK6v6L   fSSaS[#'#7#7 MM %q	' r1   rS   rW   rY   r[   s   `` r/   max_horizontalz"PandasLikeNamespace.max_horizontal   r   r1   c                J    | j                   j                         j                  S )zvConcatenate pandas objects along a particular axis.

        Return the **native** equivalent of `pd.concat`.
        )r:   ro   r   r-   s    r/   _concatzPandasLikeNamespace._concat  s     ##779@@@r1   c                   | j                   j                         r'| j                  dk  r| j                  |t        d      S | j                  |t              S )N   Fr   copyr   )r:   	is_pandas_backend_versionr   r'   r.   dfss     r/   _concat_diagonalz$PandasLikeNamespace._concat_diagonal  sI    ))+0E0E0L<<(<??||Ch|//r1   c                  | j                   j                         rQt        j                         5  t        j                  ddt
               | j                  |t              cd d d        S | j                   j                         r'| j                  dk  r| j                  |t        d      S | j                  |t              S # 1 sw Y    xY w)NignorezDThe behavior of array concatenation with empty entries is deprecated)messagecategoryr   r   Fr   )
r:   is_cudfwarningscatch_warningsfilterwarningsFutureWarningr   r(   r   r   r   s     r/   _concat_horizontalz&PandasLikeNamespace._concat_horizontal!  s     '')((* :''b*
 ||Cj|9: : !!++-$2G2G$2N<<*5<AA||Cj|11: :s   3CCc                  |d   j                   }t        |dd  d      D ]n  \  }}|j                   }t        |      t        |      k(  r||k(  j                         r=d|j	                          d| d|j	                          d}t        |       | j                  j                         r'| j                  dk  r| j                  |t        d	
      S | j                  |t              S )Nr   r$   )startz>unable to vstack, column names don't match:
   - dataframe 0: z
   - dataframe z: 
r   Fr   r   )columns	enumerater   allto_list	TypeErrorr:   r   r   r   r'   )r.   r   cols_0irE   cols_currentrz   s          r/   _concat_verticalz$PandasLikeNamespace._concat_vertical0  s    Qs12wa0 
	%EAr::L\"c&k18N7S7S7U))/)9(: ;&&'S<+?+?+A*B"F 
  n$
	% ))+0E0E0L<<(<??||Ch|//r1   c                   | j                   j                  j                         dfd}| j                  j	                  |t         t         |       S )Nc                   D cg c]  } ||       D ]  }|  }}}|D cg c]  }|j                         }}|D cg c]  }|j                          }}s=t        t        j                  |      }t        fd|      j                  | d       }|gS d t        ||      D        ^}}	|j                  |j                         j                  d|j                  j                  |j                  j                              fd|d d D        }
t        t        j                  d t        |
|	      D        |      }|gS c c}}w c c}w c c}w )Nc                    | z   |z   S r,   rB   )rM   rN   	separators     r/   rO   z>PandasLikeNamespace.concat_str.<locals>.func.<locals>.<lambda>M  s    Q]Q-> r1   c              3  H   K   | ]  \  }}|j                  | d         yw Nzip_with)rC   rD   nms      r/   rF   z?PandasLikeNamespace.concat_str.<locals>.func.<locals>.<genexpr>T  s&      ',1ArAJJsB''s    "sep)rc   rd   r   c              3  D   K   | ]  }j                  | d         ywr  r  )rC   r  	sep_arrays     r/   rF   z?PandasLikeNamespace.concat_str.<locals>.func.<locals>.<genexpr>_  s      Sbi00"b9Ss    c              3  ,   K   | ]  \  }}||z     y wr,   rB   )rC   rD   vs      r/   rF   z?PandasLikeNamespace.concat_str.<locals>.func.<locals>.<genexpr>b  s     Ftq!QUFs   )rv   r   r   r   r   r  r   _with_native__native_namespace__rp   r   rd   r   r   )rE   r3   rD   r   rP   	null_masknull_mask_resultresult
init_valuevalues
separatorsr
  rQ   r   r  strings              @r/   rR   z,PandasLikeNamespace.concat_str.<locals>.funcF  sO   +0D%%)DQADADLD.:;affVn;F;.:;;I;#)(,,	#B  >GPP%%t2 8O%'5?	5R'#
V '33335<<!"(//55(//55	 = 	 TIcrNS
LLFz*f'EF 8O? E;;s   EEErS   rW   )r;   dtypesStringr3   rZ   r   r   )r.   r  r   rQ   rR   r  s    ``` @r/   
concat_strzPandasLikeNamespace.concat_strA  s\     %%,,. 	  	D zz(("?"G95A	 ) 
 	
r1   c                j     d fd} j                   j                  |t         t                S )Nc                    	 dd l }dd ldd lm} j
                  j                  } |t        j                   fdD               }|d   j                  } |j                  fd|D        d|D cg c]  }|j                   c}i}	j                  |	g      }
j                  }j                  }|j                         }|j                  |j                   j#                  |
      ||d   j$                  j&                        }j                  |||      gS # t        $ r}d}t	        |      |d }~ww xY wc c}w )Nr   zgstruct requires pyarrow to be installed for pandas backend. Please install pyarrow: pip install pyarrowc              3  .   K   | ]  } |        y wr,   rB   r   s     r/   rF   z;PandasLikeNamespace.struct.<locals>.func.<locals>.<genexpr>}  s     /KTR/Kr   c              3  X   K   | ]!  }j                  |j                  d        # yw)T)from_pandasN)rn   r   )rC   rD   rx   s     r/   rF   z;PandasLikeNamespace.struct.<locals>.func.<locals>.<genexpr>  s"     G1"((188(6Gs   '*field_namesrb   r   )rj   rk   pyarrow.computecomputerl   r5   _align_full_broadcastr   ru   rc   make_structchunked_arrayr;   r:   ro   rp   rq   rr   r   rd   )rE   rw   pcry   rz   alignrP   rc   rD   struct_arrayr  r=   implr   result_nativerx   rQ   r.   s   `              @r/   rR   z(PandasLikeNamespace.struct.<locals>.funcp  s4   	0#$, LL66EE///KU/KKLF!9>>D)2>>GG-34QVV4L %%|n5FmmG''D))+BII		--f5Qi&&,, & M
 LLtWLUVV3  0B  "#&C/0 5s   D 4D>	D;(D66D;rS   rW   rY   r[   s   `` r/   structzPandasLikeNamespace.structo  s>    	W@ zz(("?"G95A	 ) 
 	
r1   Nc                @    |j                   }| ||      S  |||      S r,   )where)r.   whenthen	otherwiser,  s        r/   _if_then_elsez!PandasLikeNamespace._if_then_else  s(     !JJ'/uT{KU45KKr1   c                    d fd} j                   j                  |t              t                     S )Nc                    | j                        }| j                        }j                  |j                  |j                  g      }|j                        j                  ddgf   }t        |j                  j                        gS )N)methodr   r$   r   )_evaluate_single_output_exprr   r   corrilocr   r:   r;   )	rE   a_seriesb_seriesr   r5  abr3  r.   s	        r/   rR   z&PandasLikeNamespace.corr.<locals>.func  s    66q9H66q9H))8??HOO*LMC8868*//A37D )=)=t}} r1   rS   rW   rY   )r.   r9  r:  r3  rR   s   ```` r/   r5  zPandasLikeNamespace.corr  sE    		 		 zz(("?1"E9!Q?	 ) 
 	
r1   )rX   ztype[PandasLikeDataFrame])rX   ztype[PandasLikeExpr])rX   ztype[PandasLikeSeries])rX   r   )r<   r   r=   r   rX   None)rQ   r   rX   r   )r   r!   r   zIntoDType | NonerX   r   )rX   r   )rQ   r   r   boolrX   r   )rX   z._NativeConcat[NativeDataFrameT, NativeSeriesT])r   zSequence[NativeDataFrameT]rX   r   )r   z*Sequence[NativeDataFrameT | NativeSeriesT]rX   r   )rQ   r   r  strr   r<  rX   r   r,   )r-  r   r.  r   r/  zNativeSeriesT | NonerX   r   )r9  r   r:  r   r3  r   rX   r   )__name__
__module____qualname__propertyr0   r3   r5   r8   r>   r\   r   r   r   r   r   r   r   r   r   r   r   r   r  r*  r0  r5  rB   r1   r/   r*   r*   +   sp    # #       < < 
/
b


$
48
	
<
$
48
	
<

(
( A A0
2=2	20",
$,
14,
DH,
	,
\&
X +/	LL L (	L
 
L

$2
?P
	
r1   r*   c                      e Zd Zedd	 	 	 	 	 	 	 d	d       Zedd	 	 	 	 	 	 	 d
d       Zedd	 	 	 	 	 	 	 dd       Zedd	 	 	 	 	 	 	 dd       Zdd	 	 	 	 	 	 	 ddZy)_NativeConcat.)r   c                    y r,   rB   r.   objsr   r   s       r/   __call__z_NativeConcat.__call__       r1   c                    y r,   rB   rE  s       r/   rG  z_NativeConcat.__call__  s     r1   c                    y r,   rB   rE  s       r/   rG  z_NativeConcat.__call__  rH  r1   c                    y r,   rB   rE  s       r/   rG  z_NativeConcat.__call__  s     ,/r1   Nc                    y r,   rB   rE  s       r/   rG  z_NativeConcat.__call__  s     ,/r1   )rF  zIterable[NativeDataFrameT]r   r#   r   bool | NonerX   r   )rF  zIterable[NativeSeriesT]r   r#   r   rM  rX   r   )rF  *Iterable[NativeDataFrameT | NativeSeriesT]r   r%   r   rM  rX   r   )rF  rN  r   r&   r   rM  rX   z NativeDataFrameT | NativeSeriesT)r>  r?  r@  r   rG  rB   r1   r/   rC  rC    s$     ( 	
  
  UX+6?GR	    8 	
  
    /8/ 	/
 / 
*/ / !/8/ 	/
 / 
*/r1   rC  )6
__future__r   r   r   	functoolsr   	itertoolsr   typingr   r   r	   r
   r   narwhals._compliantr   narwhals._expression_parsingr   r   narwhals._pandas_like.dataframer   narwhals._pandas_like.exprr   narwhals._pandas_like.selectorsr   narwhals._pandas_like.seriesr   narwhals._pandas_like.typingr   r   narwhals._pandas_like.utilsr   narwhals._utilsr   collections.abcr   r   typing_extensionsr   r   r   narwhals.typingr   r    r!   r"   __annotations__r#   r%   r&   r'   r(   r*   rC  rB   r1   r/   <module>r`     s    "     B B . @ 5 C 9 H ? &2+7KK 
I  > qz	9 ! Y #)[01i 1) 
K H
		H
V$/H-}<= $/r1   