我已经定制了使用plotly绘制的数据点的颜色。数据点的颜色根据与之关联的标签进行分配。
但是,在设置legend
= True
之后,绘图中将不会显示所有三种颜色(在字典中定义)。
我想要
'a': 'rgb(147,112,219)(the actual color in here)',
'b': 'rgb(220,20,60)',
'c': 'rgb(0,128,0)'
将显示在图的右上角。
import pandas as pd
import plotly as plotly
import plotly.graph_objs as go
from plotly.offline import download_plotlyjs, init_notebook_mode, plot, iplot
label = ['a', 'b', 'a', 'b', 'c']
label_df = pd.DataFrame({'color': label})
color = {'a': 'rgb(147,112,219)',
'b': 'rgb(220,20,60)',
'c': 'rgb(0,128,0)'
}
cols = label_df['color'].map(color)
data = [
go.Scatter(
x=[1, 2, 3, 4, 5],
y=[1, 2, 3, 4, 5],
mode='markers',
marker=dict(size=10, color=cols)
)
]
layout = go.Layout(
hovermode='y',
showlegend=True,
barmode='stack',
title='plot'
)
fig = go.Figure(data=data, layout=layout)
plot(fig, filename='plot.html')
关于如何在情节中显示自定义图例的任何建议?
答案 0 :(得分:0)
图例的外观和结构旨在反映人物的内容。您打算在这里完成的工作很有可能,但我认为最好以略有不同的方式来完成。如果您希望所有a
标签都具有颜色'rgb(147,112,219)'
,并且对a
有多个观察结果,那么只需构建图形以准确显示该数字即可传说是有道理的。通常,每条迹线只有一个图例元素。并且您有一个跟踪。您和我的方法之间的主要区别是,我为每个唯一标签添加了一条痕迹。下面的代码段将为您提供以下图表,据我了解这是您想要的结果:
完整代码:
import pandas as pd
import plotly.graph_objects as go
df = pd.DataFrame({'label': ['a', 'b', 'a', 'b', 'c'],
'x': [1, 2, 3, 4, 5],
'y': [1, 2, 3, 4, 5]})
color = {'a': 'rgb(147,112,219)',
'b': 'rgb(220,20,60)',
'c': 'rgb(0,128,0)'}
fig = go.Figure()
for lbl in df['label'].unique():
dfp = df[df['label']==lbl]
#print(dfp)
fig.add_traces(go.Scatter(x=dfp['x'], y=dfp['y'], mode='markers',
name=lbl,
marker = dict(color=color[lbl], size = 12)
))
fig.show()