python - Plotly:如何为离散分类变量设置等值线图颜色？

Question

我正在尝试绘制一张世界地图，其中所有国家都有不同的风险水平(低、中和高)。我想为每个风险级别设置不同的颜色，但不知道如何更改配色方案，以便每个风险类别都有我选择的颜色。
df.risk 变量目前低至 1，中等至 2，高至 3，因此它是一个连续变量，但是我想使用离散的，

fig = go.Figure(data=go.Choropleth(
    locations = df['code'],
    z = df['risk'],
    text = df['COUNTRY'],
    colorscale = 'Rainbow',
    autocolorscale=False,
    reversescale=True,
    marker_line_color='darkgray',
    marker_line_width=0.5,
    colorbar_tickprefix = '',
    colorbar_title = 'Risk level',
))

fig.update_layout(
    title_text='Risk map',
    geo=dict(
        showframe=False,
        showcoastlines=False,
        projection_type='equirectangular'
    ),
    annotations = [dict(
        x=0.55,
        y=0.15,
        xref='paper',
        yref='paper',
        text='Source: <a href="www.google.com">\
            Google</a>',
        showarrow = False
    )]
)

fig.show()

我的样本df是:

{'Country': {0: 'Afghanistan',
  1: 'Albania',
  2: 'Algeria',
  3: 'American Samoa',
  4: 'Andorra'},
 'code': {0: 'AFG', 1: 'ALB', 2: 'DZA', 3: 'ASM', 4: 'AND'},
 'risk': {0: 'High', 1: 'Moderate', 2: 'High', 3: 'Low', 4: 'High'}}

Answer 1

在这种情况下，我宁愿使用 plotly.express与 color=df['risk']然后设置color_discrete_map={'High':'red', 'Moderate':'Yellow','Low':'Green'} :
阴谋:

完整代码:

import plotly.express as px
import pandas as pd

fig = px.choropleth(locations=df['Country'], 
                    locationmode="country names",
                    color=df['risk'],
                    color_discrete_map={'High':'red',
                                        'Moderate':'Yellow',
                                        'Low':'Green'}
                    #scope="usa"
                   )
fig.show()