Python 彩虹散点图

我必须用python制作以下散点图。此绘图的代码为：

n = 1024
X = np.random.normal(0,1,n)
Y = np.random.normal(0,1,n)

plt.scatter(X,Y)

---

但作为尊重，这不会给颜色。我试了很多，但找不到解决办法。我知道这和图中X/Y的角度有关，但我不知道怎么做

逻辑很可能是从origo到点的角度。这可以通过

np.arctan2（X，Y）

轻松计算。我不知道您的示例中使用了哪种颜色贴图，但您可能可以在这里找到：

在

plt.scatter

要获得与您的示例类似的内容，请执行以下操作：

---

plt.scatter（X，Y，c=np.arctan2（X，Y），cmap='rainbow'，s=50，alpha=0.8）

着色背后的逻辑是什么？@alec\u djinn这是我必须做的一个练习。。。我也不知道逻辑，只是想知道如何做到这一点。你是在问如何在maptplotlib上使用带有散点图的颜色渐变还是什么？