AHj9ddlmZddlmZmZmZddlmZddl m Z ddl m Z ddl mZddlmZddlmZer:ddlZdd lmZdd lmZdd lmZdd lmZmZej:d k\rddlm Z nddlm Z GddZ y)) annotations) TYPE_CHECKINGLiteraloverloadN) deprecated)serialize_polars_object)display_dot_graph) wrap_expr) ComputeError)IOBase)Path)Expr) SchemaDictSerializationFormat) c@eZdZdZdZd1dZd2dZd2dZd3dZd4dZ d4dZ d4d Z d4d Z d5d Z d5d Zd5d Zddd6dZddd6dZeddd7dZed8dZddd9dZddd:dZd;dZddd d?d Zed>d@d!Zedd dAd"Z dBd#d dCd$Zed>dDd%ZedEd&Zed'dBdFd(Zeddd) dGd*Zedd dHd+Zddd) dId,Zdddd-dd. dJd/ZdKd0Zy)LExprMetaNameSpacez*Namespace for expressions on a meta level.metac&|j|_yN)_pyexprselfexprs H/root/tools/cai/cai_env/lib/python3.12/site-packages/polars/expr/meta.py__init__zExprMetaNameSpace.__init__s || cNt|jjdSNz.meta)r r__str__rs rr!zExprMetaNameSpace.__str__"s"DLL)1134E::rcNt|jjdSr )r r__repr__r"s rr$zExprMetaNameSpace.__repr__%s"DLL)2245U;;rc6|jjSr)r__hash__r"s rr&zExprMetaNameSpace.__hash__(s||$$&&rcL|jj|jSrrmeta_eqrothers r__eq__zExprMetaNameSpace.__eq__+s||##EMM22rc||k( Srr*s r__ne__zExprMetaNameSpace.__ne__.s5=  rcL|jj|jS)aX Indicate if this expression is the same as another expression. Examples -------- >>> foo_bar = pl.col("foo").alias("bar") >>> foo = pl.col("foo") >>> foo_bar.meta.eq(foo) False >>> foo_bar2 = pl.col("foo").alias("bar") >>> foo_bar.meta.eq(foo_bar2) True r(r*s reqzExprMetaNameSpace.eq1s||##EMM22rc&|j| S)a\ Indicate if this expression is NOT the same as another expression. Examples -------- >>> foo_bar = pl.col("foo").alias("bar") >>> foo = pl.col("foo") >>> foo_bar.meta.ne(foo) True >>> foo_bar2 = pl.col("foo").alias("bar") >>> foo_bar.meta.ne(foo_bar2) False )r1r*s rnezExprMetaNameSpace.neAs775>!!rc6|jjS)z Indicate if this expression expands into multiple expressions. Examples -------- >>> e = pl.col(["a", "b"]).name.suffix("_foo") >>> e.meta.has_multiple_outputs() True )rmeta_has_multiple_outputsr"s rhas_multiple_outputsz&ExprMetaNameSpace.has_multiple_outputsQs||5577rc6|jjS)aq Indicate if this expression is a basic (non-regex) unaliased column. Examples -------- >>> e = pl.col("foo") >>> e.meta.is_column() True >>> e = pl.col("foo") * pl.col("bar") >>> e.meta.is_column() False >>> e = pl.col(r"^col.*\d+$") >>> e.meta.is_column() False )rmeta_is_columnr"s r is_columnzExprMetaNameSpace.is_column]s ||**,,rc6|jjS)z Indicate if this expression expands to columns that match a regex pattern. Examples -------- >>> e = pl.col("^.*$").name.prefix("foo_") >>> e.meta.is_regex_projection() True )rmeta_is_regex_projectionr"s ris_regex_projectionz%ExprMetaNameSpace.is_regex_projectionos||4466rF)allow_aliasingc8|jj|S)az Indicate if this expression only selects columns (optionally with aliasing). This can include bare columns, columns matched by regex or dtype, selectors and exclude ops, and (optionally) column/expression aliasing. .. versionadded:: 0.20.30 Parameters ---------- allow_aliasing If False (default), any aliasing is not considered to be column selection. Set True to allow for column selection that also includes aliasing. Examples -------- >>> import polars.selectors as cs >>> e = pl.col("foo") >>> e.meta.is_column_selection() True >>> e = pl.col("foo").alias("bar") >>> e.meta.is_column_selection() False >>> e.meta.is_column_selection(allow_aliasing=True) True >>> e = pl.col("foo") * pl.col("bar") >>> e.meta.is_column_selection() False >>> e = cs.starts_with("foo") >>> e.meta.is_column_selection() True >>> e = cs.starts_with("foo").exclude("foo!") >>> e.meta.is_column_selection() True )rmeta_is_column_selectionrr=s ris_column_selectionz%ExprMetaNameSpace.is_column_selection{sH||44^DDrc8|jj|S)a Indicate if this expression is a literal value (optionally aliased). .. versionadded:: 1.14 Parameters ---------- allow_aliasing If False (default), only a bare literal will match. Set True to also allow for aliased literals. Examples -------- >>> from datetime import datetime >>> e = pl.lit(123) >>> e.meta.is_literal() True >>> e = pl.lit(987.654321).alias("foo") >>> e.meta.is_literal() False >>> e = pl.lit(datetime.now()).alias("bar") >>> e.meta.is_literal(allow_aliasing=True) True )rmeta_is_literalr@s r is_literalzExprMetaNameSpace.is_literals2||++N;;rT)raise_if_undeterminedcyrr.rrEs r output_namezExprMetaNameSpace.output_namesRUrcyrr.rGs rrHzExprMetaNameSpace.output_namesSVrc\ |jjS#t$r|sYywxYw)aa Get the column name that this expression would produce. It may not always be possible to determine the output name as that can depend on the schema of the context; in that case this will raise `ComputeError` if `raise_if_undetermined` is True (the default), or `None` otherwise. Examples -------- >>> e = pl.col("foo") * pl.col("bar") >>> e.meta.output_name() 'foo' >>> e_filter = pl.col("foo").filter(pl.col("bar") == 13) >>> e_filter.meta.output_name() 'foo' >>> e_sum_over = pl.sum("foo").over("groups") >>> e_sum_over.meta.output_name() 'foo' >>> e_sum_slice = pl.sum("foo").slice(pl.len() - 10, pl.col("bar")) >>> e_sum_slice.meta.output_name() 'foo' >>> pl.len().meta.output_name() 'len' N)rmeta_output_namer rGs rrHzExprMetaNameSpace.output_names22 <<002 2 (  s  ++N)schemacp|jj|Dcgc] }t|c}Scc}w)a< Pop the latest expression and return the input(s) of the popped expression. Returns ------- list of Expr A list of expressions which in most cases will have a unit length. This is not the case when an expression has multiple inputs. For instance in a `fold` expression. Examples -------- >>> e = pl.col("foo") + pl.col("bar") >>> first = e.meta.pop()[0] >>> first.meta == pl.col("bar") True >>> first.meta == pl.col("foo") False )rmeta_popr )rrLes rpopzExprMetaNameSpace.pops+('+ll&;&;F&CD ! DDDs3c6|jjS)aD Get a list with the root column name. Examples -------- >>> e = pl.col("foo") * pl.col("bar") >>> e.meta.root_names() ['foo', 'bar'] >>> e_filter = pl.col("foo").filter(pl.col("bar") == 13) >>> e_filter.meta.root_names() ['foo', 'bar'] >>> e_sum_over = pl.sum("foo").over("groups") >>> e_sum_over.meta.root_names() ['foo', 'groups'] >>> e_sum_slice = pl.sum("foo").slice(pl.len() - 10, pl.col("bar")) >>> e_sum_slice.meta.root_names() ['foo', 'bar'] )rmeta_root_namesr"s r root_nameszExprMetaNameSpace.root_namess&||++--rcHt|jjS)aR Undo any renaming operation like `alias` or `name.keep`. Examples -------- >>> e = pl.col("foo").alias("bar") >>> e.meta.undo_aliases().meta == pl.col("foo") True >>> e = pl.col("foo").sum().over("bar") >>> e.name.keep().meta.undo_aliases().meta == e True )r rmeta_undo_aliasesr"s r undo_aliaseszExprMetaNameSpace.undo_aliases s779::rc,t|jS)zReturn the original expression.)r rr"s r as_expressionzExprMetaNameSpace.as_expressions&&rcptjj|jj S)a2 Try to turn this expression in a selector. Raises if the underlying expressions is not a column or selector. .. warning:: This functionality is considered **unstable**. It may be changed at any point without it being considered a breaking change. )plSelector_from_pyselectorr into_selectorr"s r as_selectorzExprMetaNameSpace.as_selector s&{{++DLL,F,F,HIIr.formatcyrr.rfiler`s r serializezExprMetaNameSpace.serialize,srcyrr.rbs rrdzExprMetaNameSpace.serialize1sNQrcyrr.rbs rrdzExprMetaNameSpace.serialize4srbinaryc|dk(r|jj}n,|dk(r|jj}nd|}t|t |||S)u Serialize this expression to a file or string in JSON format. Parameters ---------- file File path to which the result should be written. If set to `None` (default), the output is returned as a string instead. format The format in which to serialize. Options: - `"binary"`: Serialize to binary format (bytes). This is the default. - `"json"`: Serialize to JSON format (string). See Also -------- Expr.deserialize Notes ----- Serialization is not stable across Polars versions: a LazyFrame serialized in one Polars version may not be deserializable in another Polars version. Examples -------- Serialize the expression into a binary representation. >>> expr = pl.col("foo").sum().over("bar") >>> bytes = expr.meta.serialize() >>> type(bytes) The bytes can later be deserialized back into an `Expr` object. >>> import io >>> pl.Expr.deserialize(io.BytesIO(bytes)) rgjsonz0`format` must be one of {'binary', 'json'}, got )rserialize_binaryserialize_json ValueErrorr)rrcr` serializermsgs rrdzExprMetaNameSpace.serialize9sXX X 66J v 44JFvjQCS/ !&z4@@rcyrr.rrcs r write_jsonzExprMetaNameSpace.write_jsonos36rcyrr.rps rrqzExprMetaNameSpace.write_jsonrs=@rz;`meta.write_json` was renamed; use `meta.serialize` insteadc(|j|dS)z Write expression to json. .. deprecated:: 0.20.11 This method has been renamed to :meth:`serialize`. rir_)rdrps rrqzExprMetaNameSpace.write_jsonus~~d6~22r)return_as_stringrLcyrr.rrtrLs r tree_formatzExprMetaNameSpace.tree_formats rcyrr.rvs rrwzExprMetaNameSpace.tree_formatsrcX|jj|}|r|St|y)am Format the expression as a tree. Parameters ---------- return_as_string: If True, return as string rather than printing to stdout. schema Optionally provide a schema for the expression tree formatter. This is a mapping of column names to their data types. If provided, it may be used to enhance the tree formatting with type information. Examples -------- >>> e = (pl.col("foo") * pl.col("bar")).sum().over(pl.col("ham")) / 2 >>> e.meta.tree_format(return_as_string=True) # doctest: +SKIP N)rmeta_tree_formatprint)rrtrLss rrwzExprMetaNameSpace.tree_formats*( LL ) )& 1 H !Hr)g0@g(@)show output_path raw_outputfigsizerLcX|jj|}t|||||S)a Format the expression as a Graphviz graph. Note that Graphviz must be installed to render the visualization (if not already present, you can download it here: ``_). Parameters ---------- show Show the figure. output_path Write the figure to disk. raw_output Return dot syntax. This cannot be combined with `show` and/or `output_path`. figsize Passed to matplotlib if `show == True`. schema Optionally provide a schema for the expression tree formatter. This is a mapping of column names to their data types. If provided, it may be used to enhance the tree formatting with type information. Examples -------- >>> e = (pl.col("foo") * pl.col("bar")).sum().over(pl.col("ham")) / 2 >>> e.meta.show_graph() # doctest: +SKIP )dotr}r~rr)rmeta_show_graphr )rr}r~rrrLrs r show_graphzExprMetaNameSpace.show_graphs5Fll**62 #!   rc^t|jj|jSr)r rmeta_replace_elementrs r_replace_elementz"ExprMetaNameSpace._replace_elements ::4<<HIIr)rrreturnNone)rstr)rint)r+zExprMetaNameSpace | Exprrbool)rr)r=rrr)rE Literal[True]rr)rELiteral[False]r str | None)rErrr)rLzSchemaDict | Nonerz list[Expr])rz list[str])rr)rz pl.Selector).)rcrr`zLiteral['binary']rbytes)rcrr`zLiteral['json']rr)rcIOBase | str | Pathr`rrrr)rcIOBase | str | Path | Noner`rrzbytes | str | None)rcrrr)rcrrr)rcrrr)rtrrLNone | SchemaDictrr)rtrrLrrr)rtrrLrrr) r}rr~zstr | Path | Nonerrrztuple[float, float]rLrrr)rrrr)__name__ __module__ __qualname____doc__ _accessorrr!r$r&r,r/r1r3r6r9r<rArDrrHrPrSrVrXr^rdrqrrwrrr.rrrrsi4I$;<'3!3 " 8-$ 7=B$EL49<6DHUU VV;?@26E,.* ;' J?B+< QQ JM'4G  ,04A'/ 4A(4A$ 4A  4Al66 @@MN3O3,/$( )"   NR#0:K  +0T#'9J <)- '3$(* * ' *  * % * "*  * XJrr)! __future__rtypingrrrpolars._reexport _reexportrZpolars._utils.deprecationrpolars._utils.serderpolars._utils.variousr polars._utils.wrapr polars.exceptionsr sysior pathlibr polarsrpolars._typingrr version_infowarningstyping_extensionsrr.rrrsT"33073(*> 7"'0zJzJr