§ q-PhW3ãó<—UddlmZddlmZmZmZmZddlmZddl m Z ddl m Z er ddlmZmZmZmZgd¢Zejd„e je je je jfD¦«ŽZd ed <ed ¦«ZGd „d ee¦«Zdd„Zdd„Zdd„Zdd„Zdd„ZdS)é)Ú annotations)Ú TYPE_CHECKINGÚCallableÚGenericÚTypeVar)ÚwarnN)Úfind_stacklevel)Ú DataFrameÚExprÚ LazyFrameÚSeries)Úregister_dataframe_namespaceÚregister_expr_namespaceÚregister_lazyframe_namespaceÚregister_series_namespacec#ó$K—|] }|jV—Œ dS©N)Ú _accessors)Ú.0Úclss úJ/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/polars/api.pyú rs$èè€ÐQÐQ˜€c„nÐQÐQÐQÐQÐQÐQózset[str]Ú_reserved_namespacesÚNScó"—eZdZdZdd„Zdd „ZdS)Ú NameSpacezHEstablish property-like namespace object for user-defined functionality.ÚnameÚstrÚ namespaceútype[NS]ÚreturnÚNonecó"—||_||_dSr)Ú _accessorÚ_ns)Úselfrr s rÚ__init__zNameSpace.__init__ s€ØˆŒØˆŒˆˆrÚinstanceú NS | Nonerú NS | type[NS]cón—|€|jS| |¦«}t||j|¦«|Sr)r&Úsetattrr%)r'r)rÚ ns_instances rÚ__get__zNameSpace.__get__$s:€Ø Ð Ø”8ˆOà—h’h˜xÑ(Ô(ˆ ݘ$œ.¨+Ñ6Ô6Ð6ØÐrN)rrr r!r"r#)r)r*rr!r"r+)Ú__name__Ú __module__Ú __qualname__Ú__doc__r(r/©rrrrsB€€€€€ØRÐRððððððððððrrrrrú+type[Expr | DataFrame | LazyFrame | Series]r"úCallable[[type[NS]], type[NS]]c󇇗dˆˆfd„ }|S)z>Register custom namespace against the underlying Polars class.Úns_classr!r"có2•—‰tvrd‰›}t|¦«‚t‰‰¦«r/td‰›d‰j›dt t ¦«¬¦«t‰‰t‰|¦«¦«‰j   ‰¦«|S)Nz#cannot override reserved namespace z%Overriding existing custom namespace z (on ú))Ú stacklevel) rÚAttributeErrorÚhasattrrr0Ú UserWarningr r-rrÚadd)r8Úmsgrrs €€rr z$_create_namespace..namespace2s°ø€Ø Õ'Ð 'Ð 'Ø@¸Ð@Ð@ˆCÝ  Ñ%Ô%Ð %Ý S˜$Ñ Ô ð Ý ØV¸ÐVÐVÀSÄ\ÐVÐVÐVÝÝ*Ñ,Ô,ð ñ ô ð õ T9 T¨8Ñ4Ô4Ñ5Ô5Ð5Ø Œ×Ò˜4Ñ Ô Ð Øˆr)r8r!r"r!r4)rrr s`` rÚ_create_namespacerA-s/øø€ð ð ð ð ð ð ð ð Ðrcó6—t|tj¦«S)uï Decorator for registering custom functionality with a Polars Expr. Parameters ---------- name Name under which the functionality will be accessed. See Also -------- register_dataframe_namespace : Register functionality on a DataFrame. register_lazyframe_namespace : Register functionality on a LazyFrame. register_series_namespace : Register functionality on a Series. Examples -------- >>> @pl.api.register_expr_namespace("pow_n") ... class PowersOfN: ... def __init__(self, expr: pl.Expr) -> None: ... self._expr = expr ... ... def next(self, p: int) -> pl.Expr: ... return (p ** (self._expr.log(p).ceil()).cast(pl.Int64)).cast(pl.Int64) ... ... def previous(self, p: int) -> pl.Expr: ... return (p ** (self._expr.log(p).floor()).cast(pl.Int64)).cast(pl.Int64) ... ... def nearest(self, p: int) -> pl.Expr: ... return (p ** (self._expr.log(p)).round(0).cast(pl.Int64)).cast(pl.Int64) >>> >>> df = pl.DataFrame([1.4, 24.3, 55.0, 64.001], schema=["n"]) >>> df.select( ... pl.col("n"), ... pl.col("n").pow_n.next(p=2).alias("next_pow2"), ... pl.col("n").pow_n.previous(p=2).alias("prev_pow2"), ... pl.col("n").pow_n.nearest(p=2).alias("nearest_pow2"), ... ) shape: (4, 4) ┌────────┬───────────┬───────────┬──────────────┠│ n ┆ next_pow2 ┆ prev_pow2 ┆ nearest_pow2 │ │ --- ┆ --- ┆ --- ┆ --- │ │ f64 ┆ i64 ┆ i64 ┆ i64 │ â•žâ•â•â•â•â•â•â•â•â•ªâ•â•â•â•â•â•â•â•â•â•â•â•ªâ•â•â•â•â•â•â•â•â•â•â•â•ªâ•â•â•â•â•â•â•â•â•â•â•â•â•â•â•¡ │ 1.4 ┆ 2 ┆ 1 ┆ 1 │ │ 24.3 ┆ 32 ┆ 16 ┆ 32 │ │ 55.0 ┆ 64 ┆ 32 ┆ 64 │ │ 64.001 ┆ 128 ┆ 64 ┆ 64 │ └────────┴───────────┴───────────┴──────────────┘ )rAÚplr ©rs rrrDs€õd ˜T¥2¤7Ñ +Ô +Ð+rcó6—t|tj¦«S)uO Decorator for registering custom functionality with a Polars DataFrame. Parameters ---------- name Name under which the functionality will be accessed. See Also -------- register_expr_namespace : Register functionality on an Expr. register_lazyframe_namespace : Register functionality on a LazyFrame. register_series_namespace : Register functionality on a Series. Examples -------- >>> @pl.api.register_dataframe_namespace("split") ... class SplitFrame: ... def __init__(self, df: pl.DataFrame) -> None: ... self._df = df ... ... def by_first_letter_of_column_names(self) -> list[pl.DataFrame]: ... return [ ... self._df.select([col for col in self._df.columns if col[0] == f]) ... for f in dict.fromkeys(col[0] for col in self._df.columns) ... ] ... ... def by_first_letter_of_column_values(self, col: str) -> list[pl.DataFrame]: ... return [ ... self._df.filter(pl.col(col).str.starts_with(c)) ... for c in sorted( ... set(df.select(pl.col(col).str.slice(0, 1)).to_series()) ... ) ... ] >>> >>> df = pl.DataFrame( ... data=[["xx", 2, 3, 4], ["xy", 4, 5, 6], ["yy", 5, 6, 7], ["yz", 6, 7, 8]], ... schema=["a1", "a2", "b1", "b2"], ... orient="row", ... ) >>> df shape: (4, 4) ┌─────┬─────┬─────┬─────┠│ a1 ┆ a2 ┆ b1 ┆ b2 │ │ --- ┆ --- ┆ --- ┆ --- │ │ str ┆ i64 ┆ i64 ┆ i64 │ â•žâ•â•â•â•â•â•ªâ•â•â•â•â•â•ªâ•â•â•â•â•â•ªâ•â•â•â•â•â•¡ │ xx ┆ 2 ┆ 3 ┆ 4 │ │ xy ┆ 4 ┆ 5 ┆ 6 │ │ yy ┆ 5 ┆ 6 ┆ 7 │ │ yz ┆ 6 ┆ 7 ┆ 8 │ └─────┴─────┴─────┴─────┘ >>> df.split.by_first_letter_of_column_names() [shape: (4, 2) ┌─────┬─────┠│ a1 ┆ a2 │ │ --- ┆ --- │ │ str ┆ i64 │ â•žâ•â•â•â•â•â•ªâ•â•â•â•â•â•¡ │ xx ┆ 2 │ │ xy ┆ 4 │ │ yy ┆ 5 │ │ yz ┆ 6 │ └─────┴─────┘, shape: (4, 2) ┌─────┬─────┠│ b1 ┆ b2 │ │ --- ┆ --- │ │ i64 ┆ i64 │ â•žâ•â•â•â•â•â•ªâ•â•â•â•â•â•¡ │ 3 ┆ 4 │ │ 5 ┆ 6 │ │ 6 ┆ 7 │ │ 7 ┆ 8 │ └─────┴─────┘] >>> df.split.by_first_letter_of_column_values("a1") [shape: (2, 4) ┌─────┬─────┬─────┬─────┠│ a1 ┆ a2 ┆ b1 ┆ b2 │ │ --- ┆ --- ┆ --- ┆ --- │ │ str ┆ i64 ┆ i64 ┆ i64 │ â•žâ•â•â•â•â•â•ªâ•â•â•â•â•â•ªâ•â•â•â•â•â•ªâ•â•â•â•â•â•¡ │ xx ┆ 2 ┆ 3 ┆ 4 │ │ xy ┆ 4 ┆ 5 ┆ 6 │ └─────┴─────┴─────┴─────┘, shape: (2, 4) ┌─────┬─────┬─────┬─────┠│ a1 ┆ a2 ┆ b1 ┆ b2 │ │ --- ┆ --- ┆ --- ┆ --- │ │ str ┆ i64 ┆ i64 ┆ i64 │ â•žâ•â•â•â•â•â•ªâ•â•â•â•â•â•ªâ•â•â•â•â•â•ªâ•â•â•â•â•â•¡ │ yy ┆ 5 ┆ 6 ┆ 7 │ │ yz ┆ 6 ┆ 7 ┆ 8 │ └─────┴─────┴─────┴─────┘] )rArCr rDs rrrys€õ~ ˜T¥2¤<Ñ 0Ô 0Ð0rcó6—t|tj¦«S)uµ Decorator for registering custom functionality with a Polars LazyFrame. Parameters ---------- name Name under which the functionality will be accessed. See Also -------- register_expr_namespace : Register functionality on an Expr. register_dataframe_namespace : Register functionality on a DataFrame. register_series_namespace : Register functionality on a Series. Examples -------- >>> @pl.api.register_lazyframe_namespace("types") ... class DTypeOperations: ... def __init__(self, lf: pl.LazyFrame) -> None: ... self._lf = lf ... ... def split_by_column_dtypes(self) -> list[pl.LazyFrame]: ... return [ ... self._lf.select(pl.col(tp)) ... for tp in dict.fromkeys(self._lf.collect_schema().dtypes()) ... ] ... ... def upcast_integer_types(self) -> pl.LazyFrame: ... return self._lf.with_columns( ... pl.col(tp).cast(pl.Int64) for tp in (pl.Int8, pl.Int16, pl.Int32) ... ) >>> >>> lf = pl.LazyFrame( ... data={"a": [1, 2], "b": [3, 4], "c": [5.6, 6.7]}, ... schema=[("a", pl.Int16), ("b", pl.Int32), ("c", pl.Float32)], ... ) >>> lf.collect() shape: (2, 3) ┌─────┬─────┬─────┠│ a ┆ b ┆ c │ │ --- ┆ --- ┆ --- │ │ i16 ┆ i32 ┆ f32 │ â•žâ•â•â•â•â•â•ªâ•â•â•â•â•â•ªâ•â•â•â•â•â•¡ │ 1 ┆ 3 ┆ 5.6 │ │ 2 ┆ 4 ┆ 6.7 │ └─────┴─────┴─────┘ >>> lf.types.upcast_integer_types().collect() shape: (2, 3) ┌─────┬─────┬─────┠│ a ┆ b ┆ c │ │ --- ┆ --- ┆ --- │ │ i64 ┆ i64 ┆ f32 │ â•žâ•â•â•â•â•â•ªâ•â•â•â•â•â•ªâ•â•â•â•â•â•¡ │ 1 ┆ 3 ┆ 5.6 │ │ 2 ┆ 4 ┆ 6.7 │ └─────┴─────┴─────┘ >>> lf = pl.LazyFrame( ... data=[["xx", 2, 3, 4], ["xy", 4, 5, 6], ["yy", 5, 6, 7], ["yz", 6, 7, 8]], ... schema=["a1", "a2", "b1", "b2"], ... orient="row", ... ) >>> lf.collect() shape: (4, 4) ┌─────┬─────┬─────┬─────┠│ a1 ┆ a2 ┆ b1 ┆ b2 │ │ --- ┆ --- ┆ --- ┆ --- │ │ str ┆ i64 ┆ i64 ┆ i64 │ â•žâ•â•â•â•â•â•ªâ•â•â•â•â•â•ªâ•â•â•â•â•â•ªâ•â•â•â•â•â•¡ │ xx ┆ 2 ┆ 3 ┆ 4 │ │ xy ┆ 4 ┆ 5 ┆ 6 │ │ yy ┆ 5 ┆ 6 ┆ 7 │ │ yz ┆ 6 ┆ 7 ┆ 8 │ └─────┴─────┴─────┴─────┘ >>> pl.collect_all(lf.types.split_by_column_dtypes()) [shape: (4, 1) ┌─────┠│ a1 │ │ --- │ │ str │ â•žâ•â•â•â•â•â•¡ │ xx │ │ xy │ │ yy │ │ yz │ └─────┘, shape: (4, 3) ┌─────┬─────┬─────┠│ a2 ┆ b1 ┆ b2 │ │ --- ┆ --- ┆ --- │ │ i64 ┆ i64 ┆ i64 │ â•žâ•â•â•â•â•â•ªâ•â•â•â•â•â•ªâ•â•â•â•â•â•¡ │ 2 ┆ 3 ┆ 4 │ │ 4 ┆ 5 ┆ 6 │ │ 5 ┆ 6 ┆ 7 │ │ 6 ┆ 7 ┆ 8 │ └─────┴─────┴─────┘] )rArCr rDs rrrÛs€õD ˜T¥2¤<Ñ 0Ô 0Ð0rcó6—t|tj¦«S)a§ Decorator for registering custom functionality with a polars Series. Parameters ---------- name Name under which the functionality will be accessed. See Also -------- register_expr_namespace : Register functionality on an Expr. register_dataframe_namespace : Register functionality on a DataFrame. register_lazyframe_namespace : Register functionality on a LazyFrame. Examples -------- >>> @pl.api.register_series_namespace("math") ... class MathShortcuts: ... def __init__(self, s: pl.Series) -> None: ... self._s = s ... ... def square(self) -> pl.Series: ... return self._s * self._s ... ... def cube(self) -> pl.Series: ... return self._s * self._s * self._s >>> >>> s = pl.Series("n", [1.5, 31.0, 42.0, 64.5]) >>> s.math.square().alias("s^2") shape: (4,) Series: 's^2' [f64] [ 2.25 961.0 1764.0 4160.25 ] >>> s = pl.Series("n", [1, 2, 3, 4, 5]) >>> s.math.cube().alias("s^3") shape: (5,) Series: 's^3' [i64] [ 1 8 27 64 125 ] )rArCr rDs rrr@s€õd ˜T¥2¤9Ñ -Ô -Ð-r)rrrr5r"r6)rrr"r6)Ú __future__rÚtypingrrrrÚwarningsrÚpolars._reexportÚ _reexportrCÚpolars._utils.variousr Úpolarsr r r r Ú__all__ÚsetÚunionrÚ__annotations__rrrArrrrr4rrúrSs®ðØ"Ð"Ð"Ð"Ð"Ð"Ð"à<Ð<Ð<Ð<Ð<Ð<Ð<Ð<Ð<Ð<Ð<Ð<ØÐÐÐÐÐàÐÐÐÐÐØ1Ð1Ð1Ð1Ð1Ð1àð:Ø9Ð9Ð9Ð9Ð9Ð9Ð9Ð9Ð9Ð9Ð9Ð9ð ð ð €ð"+ ¤ØQÐQ ¤¨r¬w¸¼ ÀbÄiÐ PÐQÑQÔQð"Ððððñð  €WˆT]„]€ð ð ð ð 𠘔 ñ ô ð ð ðððð.2,ð2,ð2,ð2,ðj_1ð_1ð_1ð_1ðDb1ðb1ðb1ðb1ðJ2.ð2.ð2.ð2.ð2.ð2.r