pythonimport numpy as np# 创建虚拟数据集x = np.random.randn(1000) # 随机生成1000个x值y = np.random.randn(1000) # 随机生成1000个y值
步骤2:计算密度
接下来,我们需要计算数据点的密度。在Python中,我们可以使用numpy.histogram2d函数来实现。这将计算出在数据集的不同区域内有多少数据点。
pythonimport numpy as np# 创建虚拟数据集x = np.random.randn(1000) # 随机生成1000个x值y = np.random.randn(1000) # 随机生成1000个y值# 计算数据点的密度hist, xedges, yedges = np.histogram2d(x, y, bins=(50, 50))
步骤3:绘制等高线密度图
现在,我们有了数据密度的信息,接下来是创建等高线密度图。我们将使用Matplotlib库来完成这个任务。
pythonimport matplotlib.pyplot as plt# 创建等高线密度图plt.contourf(xedges[:-1], yedges[:-1], hist.T, levels=20, cmap='viridis')plt.colorbar()plt.xlabel('X轴')plt.ylabel('Y轴')plt.title('等高线密度图')# 显示图形plt.show()
在上述代码中,我们使用plt.contourf函数来绘制等高线密度图。xedges和yedges包含了我们计算的密度数据的边界信息,hist.T表示要绘制的数据。levels参数指定了等高线的数量,cmap参数定义了颜色映射。

完整代码如下:
pythonimport matplotlib.pyplot as pltimport numpy as npplt.rcParams['font.sans-serif'] = ['SimHei']# 创建虚拟数据集x = np.random.randn(1000) # 随机生成1000个x值y = np.random.randn(1000) # 随机生成1000个y值# 计算数据点的密度hist, xedges, yedges = np.histogram2d(x, y, bins=(50, 50))# 创建等高线密度图plt.contourf(xedges[:-1], yedges[:-1], hist.T, levels=20, cmap='viridis')plt.colorbar()plt.xlabel('X轴')plt.ylabel('Y轴')plt.title('等高线密度图')# 显示图形plt.show()
运行代码,绘制的图像如下所示:
这个图展示了虚拟数据集的密度分布,深色区域表示数据点密集,浅色区域表示数据点稀疏。
总结本文主要介绍了使用matplotlib绘制等高线密度图的方法,等高线密度图在我们的数据分析和可视化中有非常好强的适配性,希望本文可以帮到大家。
霍格沃兹测试开发学社|免费学习资料大放送,助你事半功倍!
- 公众号 - 测试人社区