plot如何根据plotly中的分类变量分

Question

我正在使用 Plotly 进行可视化。 我想制作 plot，并根据分类变量给出分数 colors。

    fig = go.Figure()
   
    fig.add_trace(go.Scatter(x=df.Predicted, y=df.Predicted,colors='Category',mode='markers',
                        
                        ))
    fig.add_trace(go.Scatter(x=df.Predicted, y=df.real ,   colors='Category'         
                      ))
    fig.show()

其中类别是我的 dataframe 中的列。我该怎么做这种图表

Answer 1

• 你暗示了我模拟的数据帧结构
• 使用Plotly表示图形对象的更高级别 API 更简单
• 曾经调用px.scatter()来生成问题中定义的跟踪。 另外在第二次调用中重命名了痕迹以确保图例清晰并使它们成为线条

import numpy as np
import pandas as pd
import plotly.express as px

df = pd.DataFrame(
    {
        "Predicted": np.sort(np.random.uniform(3, 15, 100)),
        "real": np.sort(np.random.uniform(3, 15, 100)),
        "Category": np.random.choice(list("ABCD"), 100),
    }
)

px.scatter(df, x="Predicted", y="Predicted", color="Category").add_traces(
    px.line(df, x="Predicted", y="real", color="Category")
    .for_each_trace(
        lambda t: t.update(name="real " + t.name)
    )  # make it clear in legend this is second set of traces
    .data
)