python - 类似于plotly中seaborn的hue函数？

Question

geo1 = go.Scatter(
x=geo['Year'],
y=geo['Number'],
mode='lines',
marker=dict(color=geo['Geographical region'],size=4, showscale=False),
name='geo',
showlegend=True)


data = [geo1]

layout = dict(
title='Working VISA in UK by Regions',
xaxis=dict(title='Year'),
yaxis=dict(title='Number'), showlegend=True)
fig = dict(data=data, layout=layout)
iplot(fig)

结果显示:

我想要的是在seaborn中使用与“hue”类似的函数:

如何按不同颜色的区域进行绘图编码？

Answer 1

由于您使用的是pandas，所以我建议使用Plotly Express。在本例中，代码非常简单直观:

import plotly.express as px

fig = px.line(geo, x="Year", y="Number", color="Geographical region",
              line_group="Geographical region")
fig.show()

<小时/>

或者，从 4.8 版本开始，您甚至可以通过编写以下内容来替换 pandas 的默认绘图后端:

import pandas as pd
pd.options.plotting.backend = "plotly" # just once at the beginning

geo.plot.line(x="Year", y="Number", color="Geographical region",
              line_group="Geographical region")

引用:https://plotly.com/python/plotly-express/