Jupyter/Colab中graphing.scatter_2D属性缺失报错及替代方案

问题说明 #

在Jupyter或Colab环境运行代码时，会触发AttributeError: module 'graphing' has no attribute 'scatter_2D'错误，但相同代码在Azure环境可正常执行。报错代码如下：

# 在Jupyter和Colab中报错，在Azure中正常
import graphing

# 展示结果图表
# 暂时无需关注其实现原理
graphing.scatter_2D(dataset, label_x="harness_size", 
                        label_y="boot_size",
                        trendline=lambda x: fitted_model.params[1] * x + fitted_model.params[0])

错误发生在调用graphing.scatter_2D的代码行。

替代解决方案 #

graphing是Azure环境专属的自定义工具模块，Jupyter/Colab中没有内置对应方法。可以使用通用的plotly.express库实现散点图展示，代码如下：

import plotly.express as px
px.scatter(dataset, x="harness_size", y="boot_size", title='靴码与胸背带尺寸')

如果需要保留原代码的趋势线功能，可补充拟合线绘制逻辑：

import plotly.express as px
import plotly.graph_objects as go

# 生成基础散点图
fig = px.scatter(dataset, x="harness_size", y="boot_size", title='靴码与胸背带尺寸')

# 添加趋势线
x_values = dataset["harness_size"]
y_trend = fitted_model.params[1] * x_values + fitted_model.params[0]
fig.add_trace(go.Scatter(x=x_values, y=y_trend, mode='lines', name='趋势线'))

fig.show()

内容的提问来源于stack exchange，提问作者Maria