Legend¶

fig.update_layout(
    legend=dict(
        xanchor='left',
        x=0.01,
        yanchor='top',
        y=0.99,
        orientation='h',    
    )
)

fig = go.Figure(
    data=[
        go.Scatter(x=df_germany.year, y=df_germany.gdpPercap, name="Germany"),
        go.Scatter(x=df_france.year, y=df_france.gdpPercap, name="France"),
        go.Scatter(
            x=df_averages_europe.index,
            y=df_averages_europe.gdpPercap,
            name="Europe",
            legend="legend2",
        ),
        go.Scatter(
            x=df_averages_americas.index,
            y=df_averages_americas.gdpPercap,
            name="Americas",
            legend="legend2",
        ),
    ],
    layout=dict(
        title="GDP Per Capita",
        legend={
            "title": "By country",
            "xref": "container",
            "yref": "container",
            "y": 0.65,
            "bgcolor": "Orange",
        },
        legend2={
            "title": "By continent",
            "xref": "container",
            "yref": "container",
            "y": 0.85,
            "bgcolor": "Gold",
        },
    ),
)

import pandas as pd
import plotly.graph_objs as go

df = pd.DataFrame({
    'model': ['A', 'A', 'A', 'A', 'B', 'B', 'B', 'B'],
    'period': ['AM', 'AM', 'PM', 'PM', 'AM', 'AM', 'PM', 'PM'],
    'dow': [1,2,1,2,1,2,1,2],
    'mw': [100, 120, 130, 150, 105, 125, 135,160],    
})

# Create a list of unique values for the 'model' and 'period' columns
models = df['model'].unique()
periods = df['period'].unique()

# Define a color scale for models and line style for each period
colors = ['blue', 'orange']
line_styles = ['solid', 'dash', 'dot', 'dashdot']

# Create a figure
fig = go.Figure()

# Iterate over unique values of 'model' and 'period' to create separate lines
for i, model in enumerate(models):
    color = colors[i]   
    for j, period in enumerate(periods):
        line_style = line_styles[j]
        data_subset = df.query('model == @model & period == @period')
        fig.add_trace(go.Scatter(
            x=data_subset['dow'], 
            y=data_subset['mw'], 
            mode='lines',
            line=dict(color=color, dash=line_style),
            showlegend=False,
        ))
# Add legend1. If visibility='legendonly', legend will grey out
for i, model in enumerate(models):
    color = colors[i]
    data_subset = df.query('model == @model & period == @periods[0]')
    fig.add_trace(go.Scatter(
        x=data_subset['dow'].iloc[:1], 
        y=data_subset['mw'].iloc[:1],  
        mode='lines',
        line=dict(color=color, dash='solid'),
        name=f'{model}',
        visible=True,
        legend='legend',
    ))  
# Add legend2
for j, period in enumerate(periods):
    line_style = line_styles[j]
    data_subset = df.query('model == @models[0] & period == @period')
    fig.add_trace(go.Scatter(
        x=data_subset['dow'].iloc[:1], 
        y=data_subset['mw'].iloc[:1], 
        mode='lines',
        line=dict(color='black', dash=line_style),
        name=f'{period}',
        visible=True,
        legend='legend2',
    ))        

# Update the layout to add a legend
fig.update_layout(
    legend={
        "title": 'Model',
        # "xref": "container",
        # "yref": "container",
        'x': 1.01,
        'y': 0.99,
        # "bgcolor": "Gold",
    },
    legend2={
        "title": 'Period',
        # "xref": "container",
        # "yref": "container",
        'x': 1.01,
        'y': 0.45,
        # "bgcolor": "Orange",
    },    
)

# Show the plot
fig.show()

import plotly.graph_objects as go

fig = go.Figure()

fig.add_trace(go.Scatter(
    x=[1, 2, 3],
    y=[2, 1, 3],
    legendgroup="group",  # this can be any string, not just "group"
    legendgrouptitle_text="First Group Title",
    name="first legend group",
    mode="markers",
    marker=dict(color="Crimson", size=10)
))

fig.add_trace(go.Scatter(
    x=[1, 2, 3],
    y=[2, 2, 2],
    legendgroup="group",
    name="first legend group - average",
    mode="lines",
    line=dict(color="Crimson")
))

fig.add_trace(go.Scatter(
    x=[1, 2, 3],
    y=[4, 9, 2],
    legendgroup="group2",
    legendgrouptitle_text="Second Group Title",
    name="second legend group",
    mode="markers",
    marker=dict(color="MediumPurple", size=10)
))

fig.add_trace(go.Scatter(
    x=[1, 2, 3],
    y=[5, 5, 5],
    legendgroup="group2",
    name="second legend group - average",
    mode="lines",
    line=dict(color="MediumPurple")
))

fig.update_layout(title="Try Clicking on the Legend Items!")

fig.show()