python matplotlib.pyplot颜色条
时间: 2023-07-12 20:59:34 浏览: 182
matplotlib.pyplot笔记
Matplotlib是Python中最常用的绘图库之一,可用于生成各种类型的图表。当我们在绘制图表时,可能需要添加颜色条以展示数据的变化情况。下面是一个简单的例子,展示如何添加颜色条:
```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))
# 绘制等高线
plt.contourf(X, Y, Z, 20, cmap='jet')
# 添加颜色条
plt.colorbar()
# 显示图像
plt.show()
```
在上面的代码中,我们首先生成了一些数据,然后使用`plt.contourf()`函数绘制了等高线图,并指定了颜色映射为`jet`。最后,我们使用`plt.colorbar()`函数添加了颜色条。运行代码后,将会得到一张包含颜色条的等高线图。
阅读全文