j L U d dl mZ d dlZd dlmZ d dlmZ er}d dlZd dl m
Z
d dlm Z d dl m
Z d dl mZ d d l mZ d d
l mZ d d l mZ d d l mZ d d
l mZ d dlmZ ej dk rd dlm Z nd dl m Z eez ez ez ez ez Z de d< e!e"e f Z#de d< G d d Z$dS ) ) annotationsN)
TYPE_CHECKING)altair)Callable) TypeAlias) ChannelColor) ChannelOrder) ChannelSize)ChannelTooltip)ChannelX)ChannelY)
EncodeKwds) DataFrame) )Unpackr Encoding Encodingsc Z e Zd ZdZd dZ d d dZ d d dZ d d!dZeZd"dZ dS )#
DataFramePlotzDataFrame.plot namespace.dfr returnNonec 8 t j | | _ d S )N)altChart_chart)selfr s ^/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/polars/dataframe/plotting.py__init__zDataFramePlot.__init__# s imm NxX | NoneyY | Nonecolor Color | NonekwargsUnpack[EncodeKwds] alt.Chartc i }|||d< |||d< |||d< | j d j di || S )a
Draw bar plot.
Polars does not implement plotting logic itself but instead defers to
`Altair `_.
`df.plot.bar(**kwargs)` is shorthand for
`alt.Chart(df).mark_bar().encode(**kwargs).interactive()`,
and is provided for convenience - for full customisability, use a plotting
library directly.
.. versionchanged:: 1.6.0
In prior versions of Polars, HvPlot was the plotting backend. If you would
like to restore the previous plotting functionality, all you need to do
is add `import hvplot.polars` at the top of your script and replace
`df.plot` with `df.hvplot`.
Parameters
----------
x
Column with x-coordinates of bars.
y
Column with y-coordinates of bars.
color
Column to color bars by.
**kwargs
Additional keyword arguments passed to Altair.
Examples
--------
>>> df = pl.DataFrame(
... {
... "day": ["Mon", "Tue", "Wed", "Thu", "Fri", "Sat", "Sun"] * 2,
... "group": ["a"] * 7 + ["b"] * 7,
... "value": [1, 3, 2, 4, 5, 6, 1, 1, 3, 2, 4, 5, 1, 2],
... }
... )
>>> df.plot.bar(
... x="day", y="value", color="day", column="group"
... ) # doctest: +SKIP
Nr" r$ r& Ttooltip )r mark_barencode interactive)r r" r$ r& r( encodingss r barzDataFramePlot.bar& s b " = IcN = IcN !&Ig
DK . .
+
+
+#)
+
+
[]]
r! order Order | Nonec i }|||d< |||d< |||d< |||d< | j d j di || S ) a
Draw line plot.
Polars does not implement plotting logic itself but instead defers to
`Altair `_.
`df.plot.line(**kwargs)` is shorthand for
`alt.Chart(df).mark_line().encode(**kwargs).interactive()`,
and is provided for convenience - for full customisatibility, use a plotting
library directly.
.. versionchanged:: 1.6.0
In prior versions of Polars, HvPlot was the plotting backend. If you would
like to restore the previous plotting functionality, all you need to do
is add `import hvplot.polars` at the top of your script and replace
`df.plot` with `df.hvplot`.
Parameters
----------
x
Column with x-coordinates of lines.
y
Column with y-coordinates of lines.
color
Column to color lines by.
order
Column to use for order of data points in lines.
**kwargs
Additional keyword arguments passed to Altair.
Examples
--------
>>> from datetime import date
>>> df = pl.DataFrame(
... {
... "date": [date(2020, 1, 2), date(2020, 1, 3), date(2020, 1, 4)] * 2,
... "price": [1, 4, 6, 1, 5, 2],
... "stock": ["a", "a", "a", "b", "b", "b"],
... }
... )
>>> df.plot.line(x="date", y="price", color="stock") # doctest: +SKIP
Nr" r$ r& r4 Tr, r. )r mark_liner0 r1 )r r" r$ r& r4 r( r2 s r linezDataFramePlot.lined s f " = IcN = IcN !&Ig !&Ig
DK ! !$ ! / /
+
+
+#)
+
+
[]]
r! size Size | Nonec i }|||d< |||d< |||d< |||d< | j d j di || S ) a
Draw scatter plot.
Polars does not implement plotting logic itself but instead defers to
`Altair `_.
`df.plot.point(**kwargs)` is shorthand for
`alt.Chart(df).mark_point().encode(**kwargs).interactive()`,
and is provided for convenience - for full customisatibility, use a plotting
library directly.
.. versionchanged:: 1.6.0
In prior versions of Polars, HvPlot was the plotting backend. If you would
like to restore the previous plotting functionality, all you need to do
is add `import hvplot.polars` at the top of your script and replace
`df.plot` with `df.hvplot`.
Parameters
----------
x
Column with x-coordinates of points.
y
Column with y-coordinates of points.
color
Column to color points by.
size
Column which determines points' sizes.
**kwargs
Additional keyword arguments passed to Altair.
Examples
--------
>>> df = pl.DataFrame(
... {
... "length": [1, 4, 6],
... "width": [4, 5, 6],
... "species": ["setosa", "setosa", "versicolor"],
... }
... )
>>> df.plot.point(x="length", y="width", color="species") # doctest: +SKIP
Nr" r$ r& r9 Tr, r. )r
mark_pointr0 r1 )r r" r$ r& r9 r( r2 s r pointzDataFramePlot.point s d " = IcN = IcN !&Ig $If
DK " "4 " 0 0
[]]
r! attrstrCallable[..., alt.Chart]c t | j d| d d| d}t | dd t j j D v }|rd fd
}nd fd }|S )
Nmark_zAltair has no method 'mark_'r- c h | ] }|j
S r. )name).0values r z,DataFramePlot.__getattr__.. s' 1
1
1
EJ1
1
1
r! r( r r r* c V d j di | S )NTr, r. r0 r1 r( methods r funcz'DataFramePlot.__getattr__..func s5 2vvd+++2<.func s- &vvxx0000<<>>>r! )r( r r r* )getattrr AttributeErrorinspect signature
parametersvalues)r r> msgaccepts_tooltip_argumentrM rL s @r __getattr__zDataFramePlot.__getattr__ s ndnnd;; >7777C %% %#, 1
1
$+$5f$=$=$H$O$O$Q$Q1
1
1
$
$ ?
K
K
K
K
K
K
K
?
?
?
?
?
? r! )r r r r )NNN)
r" r# r$ r% r& r' r( r) r r* )NNNN) r" r# r$ r% r& r' r4 r5 r( r) r r* ) r" r# r$ r% r& r' r9 r: r( r) r r* )r> r? r r@ )
__name__
__module__ __qualname____doc__r r3 r8 r= scatterrW r. r! r r r s ##$ $ $ $
" <
<
<
<
<
@ " " @
@
@
@
@
H " B
B
B
B
B
J G r! r )%
__future__r rQ typingr polars._dependenciesr r syscollections.abcr r
altair.typingr Colorr Orderr
Sizer Tooltipr Xr
Yr polarsr version_infor typing_extensionsr __annotations__dictr? r r r. r! r rn s " " " " " " " . . . . . . /JJJ(((((( 333333333333111111777777++++++++++++((((((
7""!!!!!!!,,,,,,a%%-%/$6@H@@@@ X
.I...._ _ _ _ _ _ _ _ _ _r!