добавление сюжетной легенды на диаграмму рассеяния с отображенными метками

Question

Я хотел бы добавить легенду к графику в этом примере стекового потока: Python - графическое назначение цветов графика рассеяния по метке

Легенда Plotly основана на каждой трассе сюжета. Но что, если у меня есть 1 трасса и различия в цвете основаны на категориальном столбце, который был сопоставлен с цветами RGB, как показано в примере.

import pandas as pd 
import plotly.plotly as py
import plotly.graph_objs as go

d  = {'x': [1, 2, 3], 'y': [3, 4, 5], 'z': ['A', 'B', 'A']}
df = pd.DataFrame(data=d)

colorsIdx = {'A': 'rgb(215,48,39)', 'B': 'rgb(215,148,39)'}
cols      = df['z'].map(colorsIdx)

# Create a trace
trace = go.Scatter(
    x = df.x,
    y = df.y,
    mode = 'markers',
    marker=dict(size=15, color=cols)
)

data = [trace]
py.iplot(data)

Я ожидаю, что легенда вывода покажет информацию в объекте colorsIdx, где мы видим точку цвета и метку.