只有一条线着色的海运多线图

2022-02-27 00:00:00 python matplotlib seaborn linechart

问题描述

我正在尝试使用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)

输出:

相关文章