如何在 Plotly 中使用 Polars 而不转换为 Pandas？

Question

我想用 Polars 替换 Pandas，但我无法找到如何在不转换为 Pandas 的情况下将 Polars 与 Plotly 一起使用。我想知道是否有一种方法可以将 Pandas 完全排除在这个过程之外。

考虑以下测试数据：

import polars as pl
import numpy as np
import plotly.express as px

df = pl.DataFrame(
    {
        "nrs": [1, 2, 3, None, 5],
        "names": ["foo", "ham", "spam", "egg", None],
        "random": np.random.rand(5),
        "groups": ["A", "A", "B", "C", "B"],
    }
)

fig = px.bar(df, x='names', y='random')
fig.show()

我希望此代码在 Jupyter 笔记本中显示条形图，但它返回错误：

/Library/Frameworks/Python.framework/Versions/3.10/lib/python3.10/site-packages/polars/internals/frame.py:1483: UserWarning: accessing series as Attribute of a DataFrame is deprecated
  warnings.warn("accessing series as Attribute of a DataFrame is deprecated")

可以使用 df = df.to_pandas 将 Polars 数据帧转换为 Pandas 数据帧()。然后，它就起作用了。然而，还有另一种更简单、更优雅的解决方案吗？

Answer 1

是的，不需要转换为 Pandas 数据帧。有人 (sa-) 请求支持更好的选项 此处 并包含了解决方法。

“我现在使用的解决方法是 px.line(x=df["a"], y=df["b"])，但如果数据框的名称太大，它会变得笨拙”

对于 OP代码示例，显式指定数据帧列的方法有效。
我发现除了用 px.bar(x=df["names"], y=df["random"]) 指定数据框列 - 或 - px.bar(df , x=df["names"], y=df["random"])，转换到列表也可以工作：

import polars as pl
import numpy as np
import plotly.express as px

df = pl.DataFrame(
    {
        "nrs": [1, 2, 3, None, 5],
        "names": ["foo", "ham", "spam", "egg", None],
        "random": np.random.rand(5),
        "groups": ["A", "A", "B", "C", "B"],
    }
)

px.bar(df, x=list(df["names"]), y=list(df["random"]))

更好地了解极坐标，一旦您看到解决方法的想法，您可能会看到一些其他选项。

发布的示例有更简单，而不是px.line(df, x="a", y="b") 就像你可以用于 Pandas 数据框一样，你使用px.line(x=df["a"], y=df["b"])。对于极坐标，即：（

import polars as pl
import plotly.express as px

df = pl.DataFrame({"a":[1,2,3,4,5], "b":[1,4,9,16,25]})

px.line(x=df["a"], y=df["b"])

请注意，使用 plotly.express 需要安装 Pandas，请参阅此处 和这里。 plotly.express 在我的答案中，因为它更接近OP。如果有需要的话，代码可以调整为使用plotly.graph_objects完全不安装 Pandas。）

Answer 2

仅供参考：plotly-express href =“ https://github.com/plotly/plotly.py.py/pull/4790” rel =“ nofollow noreferrer”>刚刚合并 DataFrame支持（通过 narwhals ），这意味着极点将是 nastily 受支持的人，因此没有更多的转换到引擎盖下的大熊猫（您可能怀疑，使用圆柱框架时会带来不错的绘图性能提升）。

Answer 3

目前可以从熊猫转移到pola.rs。从我的研究中，您的[]将起作用，但被认为是Porars中的反诉讼。作者建议您使用.to_series方法。

px.pie(df,                                   # Polars DataFrame
   names = df.select('Model').to_series(),
   values = df.select('Sales').to_series(), 
   hover_name = df.select('Model').to_series(),
   color_discrete_sequence= px.colors.sequential.Plasma_r)

httpps：> plotly-express-8DA4357D2EE0

当涉及极性数据框架的可视化时，似乎您不能完全摆脱pandas dataframe的转换。

希望这有所帮助