python - Pyplot 散点图到等值线图

Question

我正在使用 pyplot 从 python 中的数千个点制作散点图。我的问题是他们倾向于集中在一个地方，那只是一大团点。

是否有某种功能可以使 pyplot 绘图点上升直到达到某个临界密度，然后使其成为等高线图？

我的问题类似于this one ，其中示例图具有等高线，其中颜色表示绘制点的密度。

这就是我的数据的样子

Answer 1

首先，您需要估算数据的密度。根据您选择的方法，varying result可以获得。

假设您想根据 scipy.stats.gaussian_kde 的示例进行高斯密度估计 ，你可以得到密度高度:

def density_estimation(m1, m2):
    X, Y = np.mgrid[xmin:xmax:100j, ymin:ymax:100j]                                                     
    positions = np.vstack([X.ravel(), Y.ravel()])                                                       
    values = np.vstack([m1, m2])                                                                        
    kernel = stats.gaussian_kde(values)                                                                 
    Z = np.reshape(kernel(positions).T, X.shape)
    return X, Y, Z

然后，您可以使用 contour 绘制它与

X, Y, Z = density_estimation(m1, m2)

fig, ax = plt.subplots()                   

# Show density 
ax.imshow(np.rot90(Z), cmap=plt.cm.gist_earth_r,                                                    
          extent=[xmin, xmax, ymin, ymax])

# Add contour lines
plt.contour(X, Y, Z)                                                                           

ax.plot(m1, m2, 'k.', markersize=2)    

ax.set_xlim([xmin, xmax])                                                                           
ax.set_ylim([ymin, ymax])                                                                           
plt.show()

作为替代方案，您可以根据其密度更改标记颜色，如图所示 here .