Pyplot Scatter Contour plot

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所示。