Python - plotly 根据标签分配散点图颜色

Question

Python - plotly 根据标签分配散点图颜色

4

我试图在 Plotly Python 中复制 Plotly R 的基本示例之一，但发现这是不可能的。

trace0 = go.Scatter( x = x1, y = y1, mode = 'markers',
                    name = ytitle +  ' X vs ' + Atitle, 
                    marker=dict(size=4, symbol='circle', 
                                color=colorsIdx, colorbar= go.ColorBar(title= 'colorbar'),
                            colorscale='Viridis')
                    )

最接近的方法是使用色条，但这是次优的，因为我无法弄清楚如何选择颜色，使它们不会混合在一起，然后颜色图例会使图形看起来像有一些连续数据正在进行，而实际上并非如此。

- orangenarwhals

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Martin Schmelzer · Accepted Answer

使用字典将颜色进行映射：

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)