matplotlib绘制二维栅格图
时间: 2023-07-05 18:29:39 浏览: 200
MATLAB绘制栅格图
3星 · 编辑精心推荐
好的,下面是使用 matplotlib 绘制二维栅格图的代码示例:
```python
import matplotlib.pyplot as plt
import numpy as np
# 构造数据
x = np.arange(0, 10, 1)
y = np.arange(0, 10, 1)
X, Y = np.meshgrid(x, y)
Z = X**2 + Y**2
# 绘制二维栅格图
plt.imshow(Z, cmap=plt.cm.hot, origin='lower')
plt.colorbar()
plt.show()
```
解释一下代码:
首先我们使用 numpy 构造了一个网格点 (X,Y) 和对应的函数值 Z,这里我们使用 x^2 + y^2 作为一个例子。接着使用 `plt.imshow()` 函数将 Z 填充到二维栅格图中,并指定了一个热度图的颜色映射方案(`cmap=plt.cm.hot`),`origin='lower'` 表示坐标系原点在左下角。最后使用 `plt.colorbar()` 函数添加一个颜色条,`plt.show()` 函数显示图像。
运行上述代码,就可以得到一个二维栅格图,其中颜色越亮的地方表示函数值越大。
阅读全文