seaboarn散点图中的预定义点颜色

Question

我正在对seaborn进行散点图绘制，我希望这些点具有“预定义”颜色。我正在遍历数据框，并在设置hue=df['category']时使用了默认调色板。很好，但是我希望类别在每个图上都带有相同的颜色，即如果不绘制一个类别，则颜色不会改变。

我以为我可以使用类似的方法，但它似乎不起作用：

category_colour = {'Netflix':'Blue', 'TV':'Red', 'DVD':'Yellow', 'Radio':'Pink'}

plot = sns.scatterplot(x="Popularity", y="Likelihood", hue=colour, data=df)

任何帮助将不胜感激。

Answer 1

您可以尝试：

category_colour = {'Netflix':'Blue', 'TV':'Red', 'DVD':'Yellow', 'Radio':'Pink'}

plot = sns.scatterplot(x="Popularity", 
                       y="Likelihood", 
                       hue=df['category'].map(category_colour), 
                       data=df)

但请确保category_colour中的颜色有效。