python - 只有一条彩色线的 Seaborn 多线图

Question

我正在尝试使用 sns 绘制多线图，但仅将美国线保持为红色，而其他国家/地区为灰色

这是我目前所拥有的:

df = px.data.gapminder()
sns.lineplot(x = 'year', y = 'pop', data = df, hue = 'country', color = 'grey', dashes = False, legend = False)

但这不会将线条更改为灰色。我在想这之后我可以自己加上红色的美国线.....

Answer 1

你可以使用 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)

输出: