python画二维密度图
时间: 2023-08-30 19:05:30 浏览: 40
在Python中,可以使用matplotlib库来画二维密度图。以下是一个简单的例子:
```python
import numpy as np
import matplotlib.pyplot as plt
from scipy.stats import kde
# 生成数据
x = np.random.normal(size=500)
y = np.random.normal(size=500)
# 使用scipy库中的kde函数计算密度
k = kde.gaussian_kde([x, y])
# 生成网格数据
xi, yi = np.mgrid[x.min():x.max():100j, y.min():y.max():100j]
zi = k(np.vstack([xi.flatten(), yi.flatten()]))
# 绘制密度图
plt.pcolormesh(xi, yi, zi.reshape(xi.shape), shading='gouraud')
plt.show()
```
这段代码会生成一个二维正态分布随机数,并使用kde函数计算密度,最后使用pcolormesh函数绘制密度图。你可以根据需要调整生成数据的方式和绘图参数以满足你的需求。