j9ddlmZddlmZmZmZddlmZddl m Z ddl m Z ddl mZddlmZddlmZer6ddlZdd lmZdd lmZdd lmZdd lmZmZejd krddlm Z nddlm Z GddZ dS)) annotations) TYPE_CHECKINGLiteraloverloadN) deprecated)serialize_polars_object)display_dot_graph wrap_expr) ComputeError)IOBase)Path)Expr) SchemaDictSerializationFormat) c2eZdZdZdZdUdZdVd ZdVd ZdWd ZdXdZ dXdZ dXdZ dXdZ dYdZ dYdZdYdZdddZdZdddZdZeddd[d Zed\d#Zddd]d$Zd%d&d^d*Zd_d,Zd`d-Zd`d.Zdad0Ze dbd1d2dcd7Zedbddd9Zed1d2ded<Z dfd=d2dgd@ZedbdhdAZedidBZedCdfdjdDZed1d%dEdkdHZed%d&dldIZdd%dEdmdJZdd%ddKd%dLdndSZdodTZd%S)pExprMetaNameSpacez*Namespace for expressions on a meta level.metaexprrreturnNonec|j|_dSN)_pyexprselfrs U/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/polars/expr/meta.py__init__zExprMetaNameSpace.__init__s| strcTt|jdSNz.meta)r r__str__rs rr%zExprMetaNameSpace.__str__"s'DL))1133::::r!cTt|jdSr$)r r__repr__r&s rr(zExprMetaNameSpace.__repr__%s'DL))2244;;;;r!intc4|jSr)r__hash__r&s rr+zExprMetaNameSpace.__hash__(s|$$&&&r!otherExprMetaNameSpace | Exprboolc@|j|jSrrmeta_eqrr,s r__eq__zExprMetaNameSpace.__eq__+s|##EM222r!c||k Srr2s r__ne__zExprMetaNameSpace.__ne__.s5=  r!c@|j|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 r0r2s reqzExprMetaNameSpace.eq1s|##EM222r!c.|| 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 )r8r2s rnezExprMetaNameSpace.neAs775>>!!r!c4|jS)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|55777r!c4|jS)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 |**,,,r!c4|jS)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|44666r!F)allow_aliasingrDc6|j|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_selectionrrDs ris_column_selectionz%ExprMetaNameSpace.is_column_selection{sH|44^DDDr!c6|j|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_literalrGs r is_literalzExprMetaNameSpace.is_literals2|++N;;;r!T)raise_if_undeterminedrL Literal[True]cdSrr5rrLs r output_namezExprMetaNameSpace.output_namesRURUr!Literal[False] str | NonecdSrr5rOs rrPzExprMetaNameSpace.output_namesSVSVr!c^ |jS#t$r|sYdSwxYw)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 rOs rrPzExprMetaNameSpace.output_namesH2 <0022 2   ( tt  s  ,,N)schemarVSchemaDict | None list[Expr]cJd|j|DS)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 c,g|]}t|Sr5r ).0es r z)ExprMetaNameSpace.pop..sDDD ! DDDr!)rmeta_pop)rrVs rpopzExprMetaNameSpace.pops)(EDdl&;&;F&C&CDDDDr! list[str]c4|jS)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&|++---r!cNt|jS)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 s 7799:::r!c*t|jS)zReturn the original expression.)r rr&s r as_expressionzExprMetaNameSpace.as_expressions&&&r! pl.Selectorcntj|jS)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({++DL,F,F,H,HIIIr!.formatfilerqLiteral['binary']bytescdSrr5rrrrqs r serializezExprMetaNameSpace.serialize,s r!Literal['json']cdSrr5rvs rrwzExprMetaNameSpace.serialize1sNQcr!IOBase | str | PathrcdSrr5rvs rrwzExprMetaNameSpace.serialize4s sr!binaryIOBase | str | Path | Nonebytes | str | Nonec|dkr |jj}n'|dkr |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)) r|jsonz0`format` must be one of {'binary', 'json'}, got )rserialize_binaryserialize_json ValueErrorr)rrrrq serializermsgs rrwzExprMetaNameSpace.serialize9s^X X  6JJ v  4JJQvQQCS// !&z4@@@r!cdSrr5rrrs r write_jsonzExprMetaNameSpace.write_jsonos363r!cdSrr5rs rrzExprMetaNameSpace.write_jsonrs=@Sr!z;`meta.write_json` was renamed; use `meta.serialize` insteadc0||dS)z Write expression to json. .. deprecated:: 0.20.11 This method has been renamed to :meth:`serialize`. rrp)rwrs rrzExprMetaNameSpace.write_jsonus~~d6~222r!)return_as_stringrVrNone | SchemaDictcdSrr5rrrVs r tree_formatzExprMetaNameSpace.tree_formats sr!cdSrr5rs rrzExprMetaNameSpace.tree_formats cr!c`|j|}|r|St|dS)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)rrrVss rrzExprMetaNameSpace.tree_formats6( L ) )& 1 1  H !HHH4r!)g0@g(@)show output_path raw_outputfigsizerVrrstr | Path | Nonerrtuple[float, float]c^|j|}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 )dotrrrr)rmeta_show_graphr )rrrrrrVrs r show_graphzExprMetaNameSpace.show_graphs@Fl**622 #!     r!cZt|j|jSr)r rmeta_replace_elementrs r_replace_elementz"ExprMetaNameSpace._replace_elements"::4<HHIIIr!)rrrr)rr")rr))r,r-rr.)rr.)rDr.rr.)rLrMrr")rLrQrrR)rLr.rrR)rVrWrrX)rr`)rr)rri).)rrrrqrsrrt)rrrrqrxrr")rrrzrqrrrr)rrr}rqrrr~)rrrrr")rrrzrr)rrr}rrR)rrQrVrrr)rrMrVrrr")rr.rVrrrR) rr.rrrr.rrrVrrrR)rrrr)__name__ __module__ __qualname____doc__ _accessorr r%r(r+r3r6r8r:r=r@rCrHrKrrPr_rcrfrhrorwrrrrrr5r!rrrs44I$$$$;;;;<<<<''''3333!!!!3333 """" 8 8 8 8----$ 7 7 7 7=B$E$E$E$E$E$EL49<<<<<<6DHUUUUUXU VVVXV;?@26EEEEEE,....* ; ; ; ;'''' J J J J?BXQQQQXQ JMX ,04A'/ 4A4A4A4A4A4Al6666X6 @@@X@ZMNN3333ON3,/$( XNRX +0T<)- '3$(* * * * * * XJJJJJJr!r)! __future__rtypingrrrpolars._reexport _reexportrkpolars._utils.deprecationrpolars._utils.serderpolars._utils.variousr polars._utils.wrapr polars.exceptionsr sysior pathlibrpolarsrpolars._typingrr version_infowarningstyping_extensionsrr5r!rrs{""""""3333333333000000777777333333((((((****** 1JJJ>>>>>>>> 7""'''''''000000zJzJzJzJzJzJzJzJzJzJr!