python画密度散点图_如何在matplotlib中绘制以密度着色的散点图？

HUX布斯

另外，如果点数使KDE计算太慢，则可以在np.histogram2d中插入颜色：import numpy as npimport matplotlib.pyplot as pltfrom scipy.interpolate import interpndef density_scatter( x , y, ax = None, sort = True, bins = 20, **kwargs )   :    """    Scatter plot colored by 2d histogram    """    if ax is None :        fig , ax = plt.subplots()    data , x_e, y_e = np.histogram2d( x, y, bins = bins)    z = interpn( ( 0.5*(x_e[1:] + x_e[:-1]) , 0.5*(y_e[1:]+y_e[:-1]) ) , data , np.vstack([x,y]).T , method = "splinef2d", bounds_error = False )    # Sort the points by density, so that the densest points are plotted last    if sort :        idx = z.argsort()        x, y, z = x[idx], y[idx], z[idx]    ax.scatter( x, y, c=z, **kwargs )    return axif "__main__" == __name__ :    x = np.random.normal(size=100000)    y = x * 3 + np.random.normal(size=100000)    density_scatter( x, y, bins = [30,30] )