python - 在 Plotly 散点图中按颜色添加图例

Question

trace = go.Scatter(
x=x_pca_df['Principle'],
y=x_pca_df['Second'],
mode='markers',
marker=dict(color=home_data['PriceLevel'], size=4, showscale=False))

data = [trace]

layout = dict(
title='PCA Analysis',
xaxis=dict(title='First Principle Component'),
yaxis=dict(title='Second Principle Component'))
fig = dict(data=data, layout=layout)
iplot(fig)

我想在这个 plotly 图的侧面显示图例。因此，人们可以理解点的颜色代表什么。我在任何地方都找不到合适的解决方案。

Answer 1

我认为这里的问题有两个方面，首先你的 PriceLevel 是一个整数，所以 plotly 使用它作为一个尺度，这就是你指定 showscale=False 的原因。 .如果您将其更改为 showscale=True你会得到各种各样的图例，但它是一个比例尺，你需要积分。

这是第二部分；如果要将它们显示为单独的刻度，则必须将它们设置为单独的迹线。

例如

# Set up a trace for each level of PriceLevel
trace1 = go.Scatter(
x=x_pca_df.query(" PriceLevel==1")['Principle'],
y=x_pca_df.query(" PriceLevel==1")['Second'],
# Add a name for each trace to appear in the legend
name = 'PriceLevel 1', 
mode='markers',
marker=dict(color='rgba(152, 0, 0, .8)', size=4, showscale=False))

trace2 = go.Scatter(
x=x_pca_df.query(" PriceLevel==2")['Principle'],
y=x_pca_df.query(" PriceLevel==2")['Second'],
name = 'PriceLevel 2', 
mode='markers',
marker=dict(color='rgba(255, 182, 193, .9)', size=4, showscale=False))

# Join them all together 
data = [trace1, trace2]

layout = dict(
title='PCA Analysis',
xaxis=dict(title='First Principle Component'),
yaxis=dict(title='Second Principle Component'))
fig = dict(data=data, layout=layout)
iplot(fig)

希望这会奏效。