o
i @ s U d dl mZ d dlZd dlmZ d dlmZ erd 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 dkred dlm Z nd dl m Z eeB eB eB eB eB Z de d< e!e"e f Z#de d< G dd dZ$dS ) ) annotationsN)
TYPE_CHECKING)altair)Callable) TypeAlias) ChannelColor) ChannelOrder) ChannelSize)ChannelTooltip)ChannelX)ChannelY)
EncodeKwds) DataFrame) )Unpackr Encoding Encodingsc @ sb e Zd ZdZd!ddZ d"d#ddZ d$d%ddZ d$d&ddZeZd'd d Z dS )(
DataFramePlotzDataFrame.plot namespace.dfr returnNonec C s t || _d S )N)altZChart_chart)selfr r P/home/app/Keep/.python/lib/python3.10/site-packages/polars/dataframe/plotting.py__init__# s zDataFramePlot.__init__NxX | NoneyY | Nonecolor Color | NonekwargsUnpack[EncodeKwds] alt.Chartc K sV i }|dur
||d< |dur||d< |dur||d< | j jddjdi || 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" Ttooltipr )r Zmark_barencode interactive)r r r r" r$ encodingsr r r bar& s 1 zDataFramePlot.barorder Order | Nonec K f i }|dur
||d< |dur||d< |dur||d< |dur"||d< | j jddjdi || 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" r- Tr' r )r Z mark_liner) r* )r r r r" r- r$ r+ r r r lined s$ 3 zDataFramePlot.linesize Size | Nonec K r/ ) 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" r1 Tr' r )r Z
mark_pointr) r* )r r r r" r1 r$ r+ r r r point s$ 2 zDataFramePlot.pointattrstrCallable[..., alt.Chart]c sv t | jd| d d u rd| d}t|ddd t j D v }|r2d fd d }|S d fd
d }|S )NZmark_zAltair has no method 'mark_'r( c S s h | ]}|j qS r )name).0valuer r r s z,DataFramePlot.__getattr__..r$ r
r r& c s ddj di | S )NTr' r r) r* r$ methodr r func s z'DataFramePlot.__getattr__..funcc s j di | S )Nr r< r= r> r r r@ s )r$ r
r r& )getattrr AttributeErrorinspect signature
parametersvalues)r r4 msgZaccepts_tooltip_argumentr@ r r> r __getattr__ s zDataFramePlot.__getattr__)r r r r )NNN)
r r r r! r" r# r$ r% r r& )NNNN) r r r r! r" r# r- r. r$ r% r r& ) r r r r! r" r# r1 r2 r$ r% r r& )r4 r5 r r6 )
__name__
__module__ __qualname____doc__r r, r0 r3 ZscatterrH r r r r r s&
@ D Er )%
__future__r rC typingr Zpolars._dependenciesr r syscollections.abcr r Z
altair.typingr Colorr ZOrderr Sizer
ZTooltipr Xr Yr
Zpolarsr version_infor Ztyping_extensionsr __annotations__dictr5 r r r r r r s,