彩虹散点图 Python
Rainbow scatter plot Python
我必须在 python 中绘制以下散点图。该图的代码是:
n = 1024
X = np.random.normal(0,1,n)
Y = np.random.normal(0,1,n)
plt.scatter(X,Y)
但是据我所知,这不会给出颜色。我已经尝试了很多,但找不到解决方案。我知道它与情节中 X/Y 的角度有关,但无法找到如何做到这一点。
逻辑很可能是从原点到点的角度。这可以用 np.arctan2(X, Y)
轻松计算。我不知道您的示例中使用了哪个颜色图,但您可以在这里找到它:https://matplotlib.org/examples/color/colormaps_reference.html
在plt.scatter
中使用指向c
关键字的角度
要获得与您的示例类似的内容:
plt.scatter(X,Y, c=np.arctan2(X, Y), cmap='rainbow', s=50, alpha=0.8)
我必须在 python 中绘制以下散点图。该图的代码是:
n = 1024
X = np.random.normal(0,1,n)
Y = np.random.normal(0,1,n)
plt.scatter(X,Y)
但是据我所知,这不会给出颜色。我已经尝试了很多,但找不到解决方案。我知道它与情节中 X/Y 的角度有关,但无法找到如何做到这一点。
逻辑很可能是从原点到点的角度。这可以用 np.arctan2(X, Y)
轻松计算。我不知道您的示例中使用了哪个颜色图,但您可以在这里找到它:https://matplotlib.org/examples/color/colormaps_reference.html
在plt.scatter
c
关键字的角度
要获得与您的示例类似的内容:
plt.scatter(X,Y, c=np.arctan2(X, Y), cmap='rainbow', s=50, alpha=0.8)