Python 在 Plotly 中手动定义的图例

Question

我有一些按天细分的数据。对于每一天，我在一天的开始和结束时都有一个数据点，每个数据点的值都在 0 到 100 之间。我需要将这些数据显示为分组条形图，x 轴为天数，y 轴为值条形颜色由它们的值决定。对于每一天，左侧栏需要有相应的开始值，右侧栏显示当天的结束值。然而图例需要显示颜色信息而不是轨迹情节基本上需要看起来像这样，但图例需要显示“绿色”、“琥珀色”、“红色”而不是“开始”、“结束”。

I need the plot to look like this but with a legend describing the colors rather than the traces

这里是重现情节的一些代码：

x = ["day"+str(i) for i in range(1,8)]
starts = [10, 50, 70, 75, 20, 50, 90]
ends = [95, 5, 80, 20, 50, 10, 75]
starts_colors = ['green', 'orange', 'red', 'red', 'green', 'orange', 'red']
ends_colors = ['red', 'green', 'red', 'green', 'orange', 'green', 'red']

这是我为上面的情节编写的代码。

layout = go.Layout(showlegend=True) 
fig = go.Figure(layout=layout)
fig.add_trace(go.Bar(x=x, y=starts, name = 'start', marker=dict(color=starts_colors)))
fig.add_trace(go.Bar(x=x, y=ends, name = 'end', marker=dict(color=ends_colors)))
fig.show()

如果我将数据重新排列成 3 条迹线（每种颜色一条），并在开始和结束处具有相应的值，我最终会在条形之间出现间隙。例如“day1”中间会有一个空隙，因为“day1”没有橙色条。

这似乎是一个简单的问题，但我不知道如何让它按照我应该的方式工作。

Answer 1

这将准确创建您请求的图表
首先将样本数据放入数据框中以打开 Plotly Express
首先更新轨迹以使用颜色列
图例添加完成。真的不是功能图例，因为它不能用于过滤图形，只会显示图形中使用的独特颜色。这是通过添加额外的 small traces

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

df = pd.DataFrame(
    {
        "day": ["day" + str(i) for i in range(1, 8)],
        "starts": [10, 50, 70, 75, 20, 50, 90],
        "ends": [95, 5, 80, 20, 50, 10, 75],
        "starts_colors": ["green", "orange", "red", "red", "green", "orange", "red"],
        "ends_colors": ["red", "green", "red", "green", "orange", "green", "red"],
    }
)

# build figure, hover_data / customdata is used to hold colors
fig = px.bar(
    df,
    x="day",
    y=["starts", "ends"],
    barmode="group",
    hover_data={"starts_colors":False, "ends_colors":False},
)

# update colors of bars
fig.plotly_update(
    data=[
        t.update(marker_color=[c[i] for c in t.customdata])
        for i, t in enumerate(fig.data)
    ]
)

# just for display purpose, create traces so that legend contains colors.  does not connect with
# bars
fig.update_traces(showlegend=False).add_traces(
    [
        go.Bar(name=c, x=[fig.data[0].x[0]], marker_color=c, showlegend=True)
        for c in np.unique(df.loc[:,["starts_colors","ends_colors"]].values.ravel())
    ]
)

Python 在 Plotly 中手动定义的图例

Manually defined legend in Plotly on Python

legend

plotly-python