python绘制二维核密度估计图（Kernel Density Estimation)

最新推荐文章于 2024-09-20 10:56:13 发布

黑白Howard

最新推荐文章于 2024-09-20 10:56:13 发布

阅读量2k

点赞数

文章标签： python 开发语言

本文链接：https://blog.csdn.net/qq_44506213/article/details/129454497

版权

def kernelDensityEstimation():
    # Generate fake data
    df =  pd.read_csv("draw_picture/down_150.csv")

    x = df['time']
    y = df['score']

    # Calculate the point density
    xy = np.vstack([x,y])
    z = gaussian_kde(xy)(xy)

    # Sort the points by density, so that the densest points are plotted last
    idx = z.argsort()
    x, y, z = x[idx], y[idx], z[idx]

    fig, ax = plt.subplots()
    # reverse bar color
    plt.scatter(x, y,c=z, s=20, cmap='Spectral_r')
    # plt.title("up_60")
    cb = plt.colorbar(extend='both')
    cb.ax.yaxis.set_tick_params(labelright=False)
    plt.ylabel("Discomfort Score")
    plt.xlabel("duration")
    plt.show()