使用标记颜色时如何格式化图例？

Question

我想跟进这个帖子：Plotly: How to colorcode plotly graph objects bar chart using Python?。

当使用 plotly express 并指定“颜色”时，可以正确生成图例，如 vestland 帖子中所见。

这是我的快速表达代码：

data = {'x_data': np.random.random_sample((5,)),
        'y_data': ['A', 'B', 'C', 'D', 'E'],
        'c_data': np.random.randint(1, 100, size=5)
       }

df = pd.DataFrame(data=data)

fig = px.bar(df,
             x='x_data',
             y='y_data',
             orientation='h',
             color='c_data',
             color_continuous_scale='YlOrRd'
            )
fig.show()

但是在使用 go.Bar 时，图例显示不正确，如下图所示：

这是我使用图形对象的代码：

bar_trace = go.Bar(name='bar_trace',
               x=df['x_data'],
               y=df['y_data'],
               marker={'color': df['c_data'], 'colorscale': 'YlOrRd'},
               orientation='h'
              )

layout = go.Layout(showlegend=True)

fig = go.FigureWidget(data=[bar_trace], layout=layout)

fig.show()

我正在学习如何使用 FigureWidget，但它似乎无法使用 plotly express，所以我必须学习如何使用图形对象进行绘图。如何将图例链接到数据，以便它像vestland 帖子中的情节表达示例一样工作。

Answer 1

这实际上归结为了解高级 API（plotly express）的作用。当您在 px 中指定 color 时，如果它是分类的，它会为每个分类值创建一个跟踪。因此，以下两种创建图形的方法大多是等效的。图例为每条轨迹显示一个项目，而不是每种颜色。

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

df = pd.DataFrame({"x":np.linspace(0,10,10), "y":np.linspace(5,15,10), "color":np.random.choice(list("ABCD"),10)})

px.bar(df, x="x", y="y", color="color", orientation="h").show()

fig = go.Figure()
for g in df.groupby("color"):
    fig.add_trace(go.Bar(x=g[1]["x"], y=g[1]["y"], name=g[0], orientation="h"))

    
fig

基于cmets的补充

• 如果您使用FigureWidget()，则不必使用图形对象，如第二张图所示，使用plotly express创建，然后生成FigureWidget()
• 对于连续数据，正常模式是使用单个迹线和颜色条（也在第二个图中演示）。但是，如果您想要一个离散的图例，请在 c_data 中为每个值创建一个跟踪并使用 https://plotly.com/python-api-reference/generated/plotly.colors.html sample_colorscale()

import plotly.express as px
import plotly.colors
import plotly.graph_objects as go
import numpy as np
import pandas as pd

# simulate data frame...
df = pd.DataFrame(
    {
        "x_data": np.linspace(0, 10, 10),
        "y_data": np.linspace(5, 15, 10),
        "c_data": np.random.randint(0, 4, 10),
    }
)

# build a trace per value in c_data using graph objects ... correct legend !!??
bar_traces = [
    go.Bar(
        name="bar_trace",
        x=d["x_data"],
        y=d["y_data"],
        marker={
            "color": plotly.colors.sample_colorscale(
                "YlOrRd",
                d["c_data"] / df["c_data"].max(),
            )
        },
        orientation="h",
    )
    for c, d in df.groupby("c_data")
]

layout = go.Layout(showlegend=True)

fig = go.FigureWidget(data=bar_traces, layout=layout)
fig.show()

fig = px.bar(
    df,
    x="x_data",
    y="y_data",
    color="c_data",
    orientation="h",
    color_continuous_scale="YlOrRd",
)
fig = go.FigureWidget(data=fig.data, layout=fig.layout)
fig.show()