WhMddlmZddlZddlmZmZddlmZerdddlZddl m Z ddl Z ddl Z ddlmZddlZddlZddlZddlmZddlmZddlmZmZddlm Z ddl!mZm"Z"dd l#m$Z$dd l%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/e0d gZ1dId Z2dId Z3dIdZ4dIdZ5dIdZ6dIdZ7dIdZ8dIdZ9dIdZ:dIdZ;dIdZdIdZ?dIdZ@dIdZAdJdZBdKdZCdLdZDdMdZEdNd ZFdOd!ZGdPd"ZHdQd#ZIdRd$ZJdSd%ZKdTd&ZLdUd'ZMdVd(ZNdWd)ZOdXd*ZPdYd+ZQdZd,ZRd[d-ZSd\d.ZTd]d/ZUd^d0ZVd_d1ZWd`d2ZXdad3ZYdbd4ZZdcd5Z[ddd6Z\ded7Z]dfd8Z^dgd9Z_dhd:Z`did;Zadjd<Zbdkd=Zcdld>Zddmd?Zednd@Zf dodAZgdpdBZhdqdCZi drdDZj dsdEZkdtdFZldtdGZmgdHZny)u) annotationsN) TYPE_CHECKINGAny) issue_warning DataFrame) TypeGuardTypeIs)SQLFrameDataFrame)r LazyFrameSeries) IntoDataFrameTIntoLazyFrameT IntoSeriesTPandasLikeDType_1DArray _1DArrayInt_2DArray_NDArray _NumpyScalar_ShapeT fireduckscBtjjddS)z;Get Polars module (if already imported - else return None).polarsNsysmodulesgetO/var/www/html/jupyter_env/lib/python3.12/site-packages/narwhals/dependencies.py get_polarsr#/ ;;??8T **r!cBtjjddS)z;Get pandas module (if already imported - else return None).pandasNrr r!r" get_pandasr'4r$r!c`tjjddx} |jSy)zAGet modin.pandas module (if already imported - else return None).modinN)rrrr&)r)s r" get_modinr*9s)$//<|| r!cBtjjddS)z9Get cudf module (if already imported - else return None).cudfNrr r!r"get_cudfr-@ ;;??64 ((r!cBtjjddS)z9Get cupy module (if already imported - else return None).cupyNrr r!r"get_cupyr1Er.r!cBtjjddS)zGet dask_expr module (if already imported - else return None).N dask_expr)r=hasattrrErrr)dds r" get_dask_exprrHhs8 ""/GB 4L|| ;;??; --r!cBtjjddS)zr!cZt|r dt|d}t|tyy)NrTzz` to `is_pandas_series`. Hint: Instead of e.g. `is_pandas_series(ser)`, did you mean `is_pandas_series(ser.to_native())`?)is_narwhals_seriesrWrrX)serrZs r"_warn_if_narwhals_seriesr_s7#T#YK(@ @ c;' r!cttx}duxrt|jxst fdt DS)zCheck whether `df` is a pandas DataFrame without importing pandas. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. Nc3K|]J}tjj|dxduxr tjj LywN)rrr isinstancer&r).0 module_namerYmods r" z&is_pandas_dataframe..sTV   T2 24? 1 r3::// 0 1VAA)r[r'rcrany IMPORT_HOOKS)rYpdrfs` @r"is_pandas_dataframerlsP r"< R , MB 1M RUV(VSr!cttx}duxrt|jxst fdt DS)zCheck whether `ser` is a pandas Series without importing pandas. Warning: This method cannot be called on Narwhals Series. Nc3K|]J}tjj|dxduxr tjj Lywrb)rrrrcr&r)rdrerfr^s r"rgz#is_pandas_series..sTT   T2 24? / sCJJ-- . /Trh)r_r'rcrrirj)r^rkrfs` @r"is_pandas_seriesrosP S!< R , KC1K PST(TQr!ctx}duxrt|jxstfdtDS)zACheck whether `index` is a pandas Index without importing pandas.Nc3K|]J}tjj|dxduxr tjj Lywrb)rrrrcr&Index)rdreindexrfs r"rgz"is_pandas_index..sTU   T2 24? 0 ucjj.. / 0Urh)r'rcrrrirj)rsrkrfs` @r"is_pandas_indexrtsF< R , LE2881L QTU(URr!cdt|tx}duxrt||jS)zCheck whether `df` is a modin DataFrame without importing modin. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. N)r[r*rcr)rYmpds r"is_modin_dataframerws. r"; Ct + M 2s}}0MMr!cdt|tx}duxrt||jS)zCheck whether `ser` is a modin Series without importing modin. Warning: This method cannot be called on Narwhals Series. N)r_r*rcr)r^rvs r"is_modin_seriesrys. S!; Ct + K 3 0KKr!cNtx}duxrt||jS)z?Check whether `index` is a modin Index without importing modin.N)r*rcrr)rsrvs r"is_modin_indexr{s$; Ct + L 5#))0LLr!cdt|tx}duxrt||jS)zCheck whether `df` is a cudf DataFrame without importing cudf. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. N)r[r-rcr)rYr,s r"is_cudf_dataframer}s. r"J Dt + N 2t~~0NNr!cdt|tx}duxrt||jS)zCheck whether `ser` is a cudf Series without importing cudf. Warning: This method cannot be called on Narwhals Series. N)r_r-rcr)r^r,s r"is_cudf_seriesrs. S!J Dt + L 3 0LLr!cNtx}duxrt||jS)z=Check whether `index` is a cudf Index without importing cudf.N)r-rcrr)rsr,s r" is_cudf_indexrs+J Dt +   tzz1r!cptx}duxr't||jxr|jdk(S)N)r1rcndarraysize)objr0s r"is_cupy_scalarrs9 D(  sDLL )  HHMr!cdt|tx}duxrt||jS)zCheck whether `df` is a Dask DataFrame without importing Dask. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. N)r[r=rcr)rYrGs r"is_dask_dataframers/ r"$& &Bt 3 T 2r||8TTr!cdt|tx}duxrt||jS)zCheck whether `df` is a DuckDB Relation without importing DuckDB. Warning: This method cannot be called on Narwhals DataFrame/LazyFrame. N)r[r@rcDuckDBPyRelation)rYr?s r"is_duckdb_relationrs7 r" l "F4 / J F # #5r!ct|tx}duxr*t||jjj S)zCheck whether `df` is a Ibis Table without importing Ibis. Warning: This method cannot be called on Narwhals DataFrame/LazyFrame. N)r[rCrcexprtypesTable)rYrBs r" is_ibis_tablers8 r"J Dt + U 2tyy?T?T0UUr!cdt|tx}duxrt||jS)zCheck whether `df` is a Polars DataFrame without importing Polars. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. N)r[r#rcrrYpls r"is_polars_dataframer. r", Bt + L 2r||0LLr!cdt|tx}duxrt||jS)zCheck whether `df` is a Polars LazyFrame without importing Polars. Warning: This method cannot be called on Narwhals DataFrame/LazyFrame. N)r[r#rcr rs r"is_polars_lazyframer"rr!cdt|tx}duxrt||jS)zCheck whether `ser` is a Polars Series without importing Polars. Warning: This method cannot be called on Narwhals Series. N)r_r#rcr)r^rs r"is_polars_seriesr,s. S!, Bt + J 3 0JJr!cxttx}xr$t|dxrt||jS)NSchema)boolr#rFrcrrrs r"is_polars_schemar6s2 :< R YWR%:Yz#ryy?Yr!c\ttx}xrt||jSrb)rr#rcDataTypers r"is_polars_data_typer=s$ jl" # D 3 (DDr!cdt|tx}duxrt||jS)zCheck whether `ser` is a PyArrow ChunkedArray without importing PyArrow. Warning: This method cannot be called on Narwhals Series. N)r_r4rc ChunkedArray)r^pas r"is_pyarrow_chunked_arrayrAs. S!- B , QC1QQr!cdt|tx}duxrt||jS)zCheck whether `df` is a PyArrow Table without importing PyArrow. Warning: This method cannot be called on Narwhals DataFrame/LazyFrame. N)r[r4rcr)rYrs r"is_pyarrow_tablerKs. r"- B , IB1IIr!cNtx}duxrt||jSrb)r4rcScalarrrs r"is_pyarrow_scalarrUs$- B , KC1KKr!c\ttx}xrt||jSrb)rr4rcrrs r"is_pyarrow_schemarYs$ km# $ CC)CCr!c\ttx}xrt||jSrb)rr4rcrrs r"is_pyarrow_data_typer]s$ km# $ EC)EEr!cvt|ttx}duxrt||jS)zCheck whether `df` is a PySpark DataFrame without importing PySpark. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. N)r[rrMrcr)rY pyspark_sqls r"is_pyspark_dataframeras> r" ') )$6 2 r;00 1 r!crt|t ddlm}t ||Sy#t$rYywxYw)zCheck whether `df` is a PySpark Connect DataFrame without importing PySpark. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. rrF)r[rOpyspark.sql.connect.dataframer ImportErrorrcrYrs r"is_pyspark_connect_dataframernsC r"(  ?"i((   s * 66cRt|tddlm}t ||Sy)zCheck whether `df` is a SQLFrame DataFrame without importing SQLFrame. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. r) BaseDataFrameF)r[rRsqlframe.base.dataframerrc)rYrs r"is_sqlframe_dataframer~s' r"~!9"m,, r!cNtx}duxrt||jS)z=Check whether `arr` is a NumPy Array without importing NumPy.N)r8rcrarrnps r"is_numpy_arrayrs$+ Bd * Jz#rzz/JJr!c:t|xr|jdk(S)z@Check whether `arr` is a 1D NumPy Array without importing NumPy.rrndimrs r"is_numpy_array_1dr #  0388q=0r!ctx}xr3t|xr&|j|j|jSrb)r8r issubdtypedtypeintegerrs r"is_numpy_array_1d_intrs9{  1 c " 1 MM#))RZZ 0r!c:t|xr|jdk(S)z@Check whether `arr` is a 2D NumPy Array without importing NumPy.rrs r"is_numpy_array_2drrr!cNtx}duxrt||jS)zACheck whether `scalar` is a NumPy Scalar without importing NumPy.N)r8rcgeneric)scalarrs r"is_numpy_scalarrs& + Bd * Mz&"**/MMr!cbt|t|xst|xs t|S)zCheck whether `df` is a pandas-like DataFrame without doing any imports. By "pandas-like", we mean: pandas, Modin, cuDF. Warning: This method cannot be called on a Narwhals DataFrame/LazyFrame. )r[rlrwr})rYs r"is_pandas_like_dataframers.r" r " U&8&< U@QRT@UUr!cbt|t|xst|xs t|S)zCheck whether `ser` is a pandas-like Series without doing any imports. By "pandas-like", we mean: pandas, Modin, cuDF. Warning: This method cannot be called on Narwhals Series. )r_roryrr^s r"is_pandas_like_seriesrs+S! C OOC$8 ON3t||jjj t jfSrb)rr'rcapi extensionsExtensionDtyper8rrrks r"is_pandas_like_dtypersD jl" #   bff.. 0A0A B)r!cttx}xr8t||jjj xr t |dS)Nto_arrow)rr'rcrrrrFrs r" is_cudf_dtypersF :< R  % sRVV..== ? % C $r!cddlm}t||xs3t|dxs%t |xst |xs t |S)aCheck whether `native_series` can be converted to a Narwhals Series. Arguments: native_series: The object to check. Returns: `True` if `native_series` can be converted to a Narwhals Series, `False` otherwise. Examples: >>> import pandas as pd >>> import polars as pl >>> import numpy as np >>> import narwhals as nw >>> s_pd = pd.Series([1, 2, 3]) >>> s_pl = pl.Series([1, 2, 3]) >>> np_arr = np.array([1, 2, 3]) >>> nw.dependencies.is_into_series(s_pd) True >>> nw.dependencies.is_into_series(s_pl) True >>> nw.dependencies.is_into_series(np_arr) False rr __narwhals_series__)narwhals.seriesrrcrFrrr) native_seriesrs r"is_into_seriesrsS4' =&) 0 ="7 8 0 M * 0 $M 2 0 ! / r!cddlm}t||xs3t|dxs%t |xst |xs t |S)aCheck whether `native_dataframe` can be converted to a Narwhals DataFrame. Arguments: native_dataframe: The object to check. Returns: `True` if `native_dataframe` can be converted to a Narwhals DataFrame, `False` otherwise. Examples: >>> import pandas as pd >>> import polars as pl >>> import numpy as np >>> from narwhals.dependencies import is_into_dataframe >>> df_pd = pd.DataFrame({"a": [1, 2, 3], "b": [4, 5, 6]}) >>> df_pl = pl.DataFrame({"a": [1, 2, 3], "b": [4, 5, 6]}) >>> np_arr = np.array([[1, 4], [2, 5], [3, 6]]) >>> is_into_dataframe(df_pd) True >>> is_into_dataframe(df_pl) True >>> is_into_dataframe(np_arr) False rr__narwhals_dataframe__)narwhals.dataframerrcrFrrr)native_dataframers r"is_into_dataframersX4- #Y/ 6 #%= > 6 / 0 6 , - 6 $$4 5 r!c&ddlm}t||S)zCheck whether `df` is a Narwhals DataFrame. This is useful if you expect a user to pass in a Narwhals DataFrame directly, and you want to catch both `narwhals.DataFrame` and `narwhals.stable.v1.DataFrame`. rr)rrrcrs r"rUrU'- b) $$r!c&ddlm}t||S)zCheck whether `lf` is a Narwhals LazyFrame. This is useful if you expect a user to pass in a Narwhals LazyFrame directly, and you want to catch both `narwhals.LazyFrame` and `narwhals.stable.v1.LazyFrame`. r)r )rr rc)lfr s r"rVrV5rr!c&ddlm}t||S)zCheck whether `ser` is a Narwhals Series. This is useful if you expect a user to pass in a Narwhals Series directly, and you want to catch both `narwhals.Series` and `narwhals.stable.v1.Series`. rr )rrrc)r^rs r"r]r]Cs' c6 ""r!cPt|xr|jjSrb)r]r is_integerrs r"is_narwhals_series_intrOs c " =syy';';'==r!)r-rCr*r8r'r#r4r}rrrrrrwryrUrVr]rrlrtrrrorrrrr)returnr)rYrrNone)r^rrr)rYrrzTypeIs[pd.DataFrame])r^rrzTypeIs[pd.Series[Any]])rsrrzTypeIs[pd.Index[Any]])rYrrzTypeIs[mpd.DataFrame])r^rrzTypeIs[mpd.Series])rsrrzTypeIs[mpd.Index[Any]])rYrrzTypeIs[cudf.DataFrame])r^rrzTypeIs[cudf.Series[Any]])rsrrzTypeIs[cudf.Index])rrrr)rYrrzTypeIs[dd.DataFrame])rYrrzTypeIs[duckdb.DuckDBPyRelation])rYrrzTypeIs[ibis.Table])rYrrzTypeIs[pl.DataFrame])rYrrzTypeIs[pl.LazyFrame])r^rrzTypeIs[pl.Series])rrrzTypeIs[pl.Schema])rrrzTypeIs[pl.DataType])r^rrzTypeIs[pa.ChunkedArray[Any]])rYrrzTypeIs[pa.Table])rrrzTypeIs[pa.Scalar[Any]])rrrzTypeIs[pa.Schema])rrrzTypeIs[pa.DataType])rYrrzTypeIs[pyspark_sql.DataFrame])rYrrzTypeIs[PySparkConnectDataFrame])rYrrzTypeIs[SQLFrameDataFrame])rzAny | _NDArray[_ShapeT]rzTypeIs[_NDArray[_ShapeT]])rrrzTypeIs[_1DArray])rrrzTypeIs[_1DArrayInt])rrrzTypeIs[_2DArray])rrrzTypeGuard[_NumpyScalar])rYrrr)r^rrr)rsrrr)rrrzTypeIs[PandasLikeDType])rrrz(TypeIs[pd.api.extensions.ExtensionDtype])rzAny | IntoSeriesTrzTypeIs[IntoSeriesT])rzAny | IntoDataFrameTrzTypeIs[IntoDataFrameT])rYzDataFrame[IntoDataFrameT] | Anyrz!TypeIs[DataFrame[IntoDataFrameT]])rzAny | LazyFrame[IntoLazyFrameT]rz!TypeIs[LazyFrame[IntoLazyFrameT]])r^zAny | Series[IntoSeriesT]rzTypeIs[Series[IntoSeriesT]])o __future__rrtypingrrnarwhals._exceptionsrr,dask.dataframe dataframerGr?rB modin.pandasr&rvrkrrr3r pyspark.sqlsqlrrrPySparkConnectDataFrametyping_extensionsr r narwhals._spark_like.dataframer rr rrnarwhals.typingrrrrrrrrrr frozensetrjr#r'r*r-r1r4r8r;r=r@rCrHrKrMrOrRr[r_rlrortrwryr{r}rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrUrVr]r__all__r r!r"rs# %.%R3@7&   "+' + + ) ) , * ) 3 + ) ., 0 8 - ((  NLM OMU VMMKERJLDF   K 1 1 N V P  -"J"J %' %& % %' %& % #> r!