matplotlib colorbar
时间: 2023-11-29 10:46:21 浏览: 43
matplotlib中的colorbar是用于显示颜色图例的工具,通常用于显示图像中每个颜色对应的数值范围。下面是两种使用colorbar的方法:
1.使用pyplot.colorbar()函数
```python
import matplotlib.pyplot as plt
import numpy as np
# 生成数据
x = np.linspace(-10, 10, 100)
y = np.linspace(-10, 10, 100)
X, Y = np.meshgrid(x, y)
Z = np.sin(np.sqrt(X**2 + Y**2))
# 绘制图像
fig, ax = plt.subplots()
im = ax.imshow(Z, cmap='coolwarm')
cbar = plt.colorbar(im)
# 显示图像
plt.show()
```
2.使用Figure.colorbar()方法
```python
import matplotlib.pyplot as plt
import numpy as np
# 生成数据
x = np.linspace(-10, 10, 100)
y = np.linspace(-10, 10, 100)
X, Y = np.meshgrid(x, y)
Z = np.sin(np.sqrt(X**2 + Y**2))
# 绘制图像
fig, ax = plt.subplots()
im = ax.imshow(Z, cmap='coolwarm')
cbar = fig.colorbar(im)
# 显示图像
plt.show()
```