plt.contourf() 使用RGB值给等值线设置颜色
时间: 2023-12-03 21:05:41 浏览: 177
`plt.contourf()`函数可以使用RGB值给等值线设置颜色,可以通过传入一个三维数组来实现。具体操作可以按照下面的步骤进行:
1. 创建一个三维数组,每个元素表示一个像素点的RGB值。
2. 通过`plt.contourf()`函数绘制等值线,其中`cmap`参数设置为`None`,表示不使用默认的颜色映射。
3. 通过`plt.imshow()`函数将三维数组作为图像显示。
示例代码如下:
```python
import numpy as np
import matplotlib.pyplot as plt
# 创建一个三维数组,每个元素表示一个像素点的RGB值
img = np.zeros((100, 100, 3), dtype=np.uint8)
for i in range(100):
for j in range(100):
img[i, j, :] = [i, j, 255]
# 绘制等值线
x = np.linspace(-1, 1, 100)
y = np.linspace(-1, 1, 100)
X, Y = np.meshgrid(x, y)
Z = np.sin(X**2 + Y**2)
plt.contourf(X, Y, Z, cmap=None)
# 显示图像
plt.imshow(img)
plt.show()
```
这段代码首先创建了一个三维数组`img`,其中每个元素表示一个像素点的RGB值。在绘制等值线时,将`cmap`参数设置为`None`,表示不使用默认的颜色映射,然后通过`plt.imshow()`函数将三维数组作为图像显示。最终的效果是,等值线使用RGB值进行绘制,图像显示了一个颜色渐变的矩形。
阅读全文