python 核密度估计

最新推荐文章于 2024-09-04 12:20:50 发布

horizon012

最新推荐文章于 2024-09-04 12:20:50 发布

阅读量772

点赞数

文章标签： python 开发语言

本文链接：https://blog.csdn.net/horizon012/article/details/133222114

版权

文章介绍了如何在Python中使用numpy和matplotlib库进行核密度估计（如Gaussian_kde）以及绘制二维数据的概率密度函数。首先展示了如何通过给定斜率k生成并可视化数据点，然后使用Scipy库进行核密度估计，最后用contourf函数绘制出概率密度函数图例。

摘要由CSDN通过智能技术生成

import numpy as np
import matplotlib.pyplot as plt

# 设置斜率 k
k = 2

# 生成 x 值，可以是一个范围内的随机数
x = np.linspace(0, 10, 1000)
# 生成对应的 y 值，按照 y = kx 的关系计算
y = k * x + np.random.normal(5, 2, 1000)  # 添加一些随机噪声

#核密度估计
xy = np.stack([x, y],axis=0) 
z = stats.gaussian_kde(xy)(xy)
idx = z.argsort() #获取索引
x, y, z = x[idx], y[idx], z[idx] 

fig,ax=plt.subplots(figsize=(7,6),dpi=300)
scatter=ax.scatter(x, y, marker='o', c=z, edgecolors=None ,s=20, label='LST',cmap='Spectral_r')

cbar=plt.colorbar(scatter,shrink=1,orientation='vertical',label='colorarbar')
# 绘制散点图

plt.xlabel('X')
plt.ylabel('Y')
plt.grid(True)
plt.show()

在这里插入图片描述

import numpy as np
from scipy import stats

# 创建一个包含二维数据的示例数组
data = np.random.randn(2, 1000)

# 使用 stats.gaussian_kde 进行核密度估计
kde = stats.gaussian_kde(data)

# 生成一组用于绘制估计概率密度函数的点
x = np.linspace(min(data[0]), max(data[0]), 100)
y = np.linspace(min(data[1]), max(data[1]), 100)
X, Y = np.meshgrid(x, y)
positions = np.vstack([X.ravel(), Y.ravel()])
Z = np.reshape(kde(positions).T, X.shape)

# 绘制估计的概率密度函数
import matplotlib.pyplot as plt
plt.contourf(X, Y, Z, cmap='Blues')
plt.xlabel('X1')
plt.ylabel('X2')
plt.title('Kernel Density')
plt.colorbar()
plt.show()