只有一条线着色的海运多线图
问题描述
我正在尝试使用SNS绘制多线图,但仅将美国线保持为红色,而其他国家为灰色
这是我到目前为止所拥有的:
df = px.data.gapminder()
sns.lineplot(x = 'year', y = 'pop', data = df, hue = 'country', color = 'grey', dashes = False, legend = False)
但这不会将线条更改为灰色。我在想,在这之后,我可以自己加一条美国线,红色的……
解决方案
您可以使用 pandas Groupby进行绘图:
fig,ax=plt.subplots()
for c,d in df.groupby('country'):
color = 'red' if c=='US' else 'grey'
d.plot(x='year',y='pop', ax=ax, color=color)
ax.legend().remove()
输出:
或者您可以将特定调色板定义为词典:
palette = {c:'red' if c=='US' else 'grey' for c in df.country.unique()}
sns.lineplot(x='year', y='pop', data=df, hue='country',
palette=palette, legend=False)
输出:
相关文章