彩虹散点图Python

时间:2018-02-01 15:08:16

标签: python scatter-plot

我必须在python中制作以下散点图。该图的代码是:

malloc

Scatterplot

但正如所料,这不会给出颜色。我已经尝试了很多,但无法找到解决方案。我知道它与情节中的X / Y角度有关,但无法找到如何做到这一点。

1 个答案:

答案 0 :(得分:2)

逻辑很可能是从原点到点的角度。这可以使用np.arctan2(X, Y)轻松计算。我不知道您的示例中使用了哪个色彩映射表,但您可以在此处找到它:https://matplotlib.org/examples/color/colormaps_reference.html

将点的角度用于c

中的plt.scatter关键字

要获得与您的示例相似的内容: plt.scatter(X,Y, c=np.arctan2(X, Y), cmap='rainbow', s=50, alpha=0.8)