python - 没有归一化的 matplotlib 颜色图

Question

我正在创建具有整数目标的数据散点图。自然，我将目标表示为散点图中的颜色。然而，有时我的模型，由于模型的性质，预测不在原始集合中的目标。即，我的训练目标是从[0,1,2]中选择的，我的模型偶尔会预测为3，因为它不是很亮。

问题是，当我对我的数据进行散点图，然后分别对预测进行散点图时，目标 2 在每个散点图中被映射到不同的颜色，这会导致图片不好。这是因为 matplotlib 默认将我的颜色列表中的值缩放到某些给定值之间。我想覆盖此默认行为，并让我的颜色列表（它们是整数）始终映射到相同的颜色，例如 1 映射到绿色，无论我的c=targets参数中有多少不同的类。

标签： pythonmatplotlib

Answer 1

创建一个颜色列表，例如colors = ['blue', 'red', 'green', 'purple']，它具有与不同目标一样多的颜色。然后，设置c=colors[target]为target模型弹出的整数。这意味着您需要一次绘制每个点，除非您对所有目标进行排序并在最后绘制。