plt.contour设置颜色
时间: 2023-09-29 18:08:33 浏览: 151
在 `plt.contour()` 函数中,可以使用参数 `colors` 来设置等高线的颜色。`colors` 可以接受以下几种形式的输入:
- 单个颜色字符串,如 `'red'`,则所有等高线都会使用该颜色。
- 一个颜色列表,如 `['red', 'green', 'blue']`,则每个等高线将按顺序使用列表中的颜色。
- 一个二维数组,如 `[[0, 'red'], [0.5, 'green'], [1, 'blue']]`,则对于每个等高线,它将根据它的高度值落在哪个颜色区间来使用相应的颜色。
例如,以下代码将绘制三个等高线,分别使用红色、绿色和蓝色:
```python
import numpy as np
import matplotlib.pyplot as plt
x = np.linspace(-1, 1, 100)
y = np.linspace(-1, 1, 100)
X, Y = np.meshgrid(x, y)
Z = X**2 + Y**2
colors = ['red', 'green', 'blue']
plt.contour(X, Y, Z, colors=colors)
plt.show()
```
如果你想让每个等高线的颜色根据高度值而变化,可以使用一个二维数组来定义颜色:
```python
colors = [[0, 'red'], [0.5, 'green'], [1, 'blue']]
plt.contour(X, Y, Z, colors=colors)
plt.show()
```
这里我们定义了三个颜色区间:从 0 到 0.5 之间使用红色到绿色渐变的颜色,从 0.5 到 1 之间使用绿色到蓝色渐变的颜色。