python - 在 Plotly 的 Scattergeo 图中制作图例

Question

我正在绘制一些类似于找到的第一个示例的数据 here (美国机 field 图)。但是，我没有绘制比例尺，而是绘制二元特征(假设一种颜色超过 15k 次飞行，一种颜色低于 15k 次飞行)。我看过文档，但如果我想做这种 plotly ，我找不到做图例的方法。有谁知道吗？

Answer 1

您可以根据自己的情况指定颜色，例如

color = np.where(df['Set'] > 15000, 'red', 'green')

但这样你就不会有一个好的传说。

另一种方法是添加两个图，每个图对应一个条件。

import pandas as pd
import plotly

df = pd.read_csv('https://raw.githubusercontent.com/plotly/datasets/master/2011_february_us_airport_traffic.csv')

data = dict(
        type = 'scattergeo',
        locationmode = 'USA-states', 
        mode = 'markers'
        )

data_high = data.copy()
data_high['lon'] = df[df['cnt']  > 10000 ]['long']
data_high['lat'] = df[df['cnt']  > 10000 ]['lat']
data_high['marker'] = dict(color = 'red')
data_high['name'] = '> 10000'

data_low = data.copy()
data_low['lon'] = df[df['cnt']  <= 10000 ]['long']
data_low['lat'] = df[df['cnt']  <= 10000 ]['lat']
data_low['marker'] = dict(color = 'green')
data_low['name'] = '<= 10000'

layout = dict(
        geo = dict(
            scope = 'usa',
            projection = dict(type='albers usa'),
        ),
    )

fig = dict(data=[data_high, data_low], layout=layout)
plotly.offline.plot(fig)