彩虹散点图Python

Question

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

malloc

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

Answer 1

逻辑很可能是从原点到点的角度。这可以使用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)