Question

我正在尝试使用Figure工厂将回归线添加到绘图中。

我的代码：

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

hover_data = "lang_code"
score_to_track = "las"

for feature_idx, feature_name in enumerate(FEATURE_NAMES):
    # declare the plot with Figure factory
    fig = go.Figure(
        layout={
            'legend': {
                'orientation': 'v'
            },
            'xaxis_title': feature_name,
            'yaxis_title': score_to_track
        }
    )

    # adding las points
    fig.add_trace(
        go.Scatter(
            x=df[feature_name],
            y=df[score_to_track],
            name=feature_name,
            text=df[hover_data],
            mode="markers+text",
            marker=dict(
                size=20,
                color="yellow"
            )
        )
    )

    # fixme : adding line
    tmp = px.scatter(df, x=feature_name, y=score_to_track, trendline="ols")
    trendline = tmp.data[1]
    fig.add_trace(trendline)

    fig.show()

对数据结构没有什么幻想，df是一个熊猫数据帧，FEATURE_NAMES是一个字符串列表。

实际上，我的代码正在运行，但是不可能：

（1）为此找到了一个更漂亮，更快速的解决方案？

（2）自定义行的颜色或名称吗？

（3）就像我使用seaborn的regplot函数一样，使用Figure factory绘制置信区间。（ref：https://seaborn.pydata.org/generated/seaborn.regplot.html）

绘制，在图上添加回归线

0 个答案: