Plotly.py 6.0.0版本中px.pie函数参数冲突问题解析

在最新发布的Plotly.py 6.0.0版本中，开发者发现了一个值得注意的问题：当同时使用names和category_order参数创建饼图时，会触发polars.exceptions.ColumnNotFoundError异常。这个问题源于内部数据处理逻辑的一个硬编码错误。

问题现象

当开发者尝试使用以下代码创建饼图时：

import plotly.express as px
import polars as pl

df = pl.DataFrame({
    "status": ["On Route", "Pending", "Waiting Result", "Delivered"],
    "count": [28, 10, 73, 8]
})

fig = px.pie(
    data_frame=df,
    values="count",
    names="status",
    category_orders={"status": ["Pending", "Waiting Result", "On Route", "Delivered"]}
)

系统会抛出ColumnNotFoundError异常，提示找不到名为"b"的列。这个错误直接影响了需要使用分类排序功能的饼图绘制场景。

技术分析

深入查看Plotly.py的源代码，可以发现问题的根源在于process_dataframe_pie函数中的硬编码列名。该函数原本应该使用names参数指定的列名进行数据处理，但却错误地使用了硬编码的"b"作为列名。

具体来说，在数据处理过程中，函数尝试创建一个临时列用于排序，但在指定列名时没有正确引用用户提供的names参数，而是直接使用了"b"这个硬编码值。这导致了当数据框中不存在"b"列时，Polars引擎抛出列未找到异常。

临时解决方案

在官方修复发布前，开发者可以采用以下几种临时解决方案：

1. 版本降级：将Plotly.py版本固定在5.24.1，这是最后一个没有此问题的版本

   pip install plotly==5.24.1
   

2. 预处理数据：在传入px.pie前手动对数据框进行排序，并禁用内部排序

   df = df.sort("status")  # 按需要的顺序预处理排序
   fig = px.pie(df, values="count", names="status")
   fig.update_traces(sort=False)  # 禁用内部排序
   

3. 使用底层API：直接使用graph_objects的Pie类创建图表

   import plotly.graph_objects as go
   fig = go.Figure(go.Pie(
       values=df["count"],
       labels=df["status"],
       sort=False
   ))
   

问题修复

核心开发团队已经确认了这个问题，并定位到了具体的错误提交。修复方案是将硬编码的"b"替换为正确的names参数引用。预计这个问题将在下一个补丁版本中得到解决。

最佳实践建议

对于依赖Plotly进行数据可视化的项目，建议：

1. 在升级到6.x版本前，充分测试饼图功能
2. 考虑在CI/CD流程中加入针对此场景的测试用例
3. 关注Plotly的版本更新日志，及时获取修复信息

这个问题虽然影响范围有限，但提醒我们在使用数据可视化库时，需要注意参数之间的潜在冲突，特别是在涉及数据排序和分类的场景下。保持对库版本变化的关注，并建立适当的测试机制，可以帮助我们及早发现和规避类似问题。