Python 是否有一种简单的方法可以将属性(例如 colored颜色、悬停名称)赋予具有情节线功能的时间序列

发布于04月23日

在下面的示例中，使用plotly.快线功能，有没有一种方法可以根据"大陆"行给线上色？把这些国家的名字作为悬停名字？

谢谢你的回答.

import numpy as np
import plotly.express as px

a=['Afghanistan','Albania','Algeria','Andorra','Angola','Antigua','Argentina','Armenia']
b=np.random.random((100,8))
c=['Asia','Europe','Africa','Europe','Africa','America','America','Asia']

df=pd.DataFrame(columns=a, data=b)
df.loc['Continent'] = c

fig = px.line(df, x=df.index, y=df.columns)
fig.show()

推荐答案

问题中的数据格式较宽，需要更改为长格式.因此，创建一个空数据框，并垂直合并大陆名称、国家名称和值数据框.然后将px.line中的 colored颜色设置设置为大陆名称.国家名称也将显示在该结果的悬停处.

import numpy as np
import pandas as pd
import plotly.express as px

a=['Afghanistan','Albania','Algeria','Andorra','Angola','Antigua','Argentina','Armenia']
b=np.random.random((100,8))
c=['Asia','Europe','Africa','Europe','Africa','America','America','Asia']

df=pd.DataFrame(columns=a, data=b)
df.loc['Continent'] = c

dff = pd.DataFrame()
for c in df.columns:
    data = pd.DataFrame({'continent':df.loc['Continent':,c][0], 'country':[c]*(len(df)-1), 'value': df.loc[0:(len(df)-2),c]})
    dff = pd.concat([dff, data], axis=0)

fig = px.line(dff, x=dff.index,
              y='value',
              color='continent',
              line_group='country',
              color_discrete_sequence=px.colors.qualitative.G10)
fig.show()